Bir modelin katmanlarını dondur
Önceden eğitilmiş ağırlıkları yükledikten sonra yeni bir görevde bir modeli ince ayar yapmaya hazırsın. Model üç adet doğrusal (linear) katman içeriyor. Ancak veri kümen küçük olduğu için bu modelin sadece son doğrusal katmanını eğitmek ve ilk iki doğrusal katmanı dondurmak istiyorsun.
Model zaten oluşturuldu ve model değişkeninde bulunuyor. Modelin parametrelerini listelemek için modelin named_parameters metodunu kullanacaksın. Her parametre bir adla tanımlanır. Bu ad, şu adlandırma kuralına uyan bir stringdir: x.name; burada x katmanın indeksidir.
Unutma, bir doğrusal katmanın iki parametresi vardır: weight ve bias.
Bu egzersiz
PyTorch ile Deep Learning'e Giriş
kursunun bir parçasıdırEgzersiz talimatları
- Bir parametrenin birinci veya ikinci katmanın ağırlığı olup olmadığını kontrol etmek için bir
ififadesi kullan. - Bu modelin ilk iki katmanının ağırlıklarını dondur.
Uygulamalı interaktif egzersiz
Bu örnek kodu tamamlayarak bu egzersizi bitirin.
for name, param in model.named_parameters():
# Check for first layer's weight
if name == '____':
# Freeze this weight
param.____ = ____
# Check for second layer's weight
if name == '____':
# Freeze this weight
param.____ = ____