unfreeze کردن لایه های آخر مدل

Question

سلام.

چطوری در مدل مدنظرمان به فرض n لایه نهایی را از فریز در بیاریم؟

Answer 1

برای باز کردن n لایه آخر در مدل PyTorch، می توانید از کد زیر استفاده کنید:

for param in model.parameters():
    param.requires_grad = False

counter = 0
for child in model.children():
    counter += 1
    if counter >= len(list(model.children())) - n:
        for param in child.parameters():
            param.requires_grad = True

 

این کد ویژگی requires_grad تمام پارامترهای مدل را در ابتدا روی False قرار می دهد. سپس، روی فرزندان مدل تکرار می‌شود و برای n لایه آخر، ویژگی requires_grad را روی True قرار می‌دهد.