fixed optimizer variable name in 15_transer_learning

patrickloeber · patrickloeber · commit c8a042717f03 · 2020-06-02T10:07:00.000+02:00
diff --git a/15_transfer_learning.py b/15_transfer_learning.py
@@ -153,7 +153,7 @@ def train_model(model, criterion, optimizer, scheduler, num_epochs=25):
 #     validate(...)
 #     scheduler.step()
 
-step_lr_scheduler = lr_scheduler.StepLR(optimizer_ft, step_size=7, gamma=0.1)
+step_lr_scheduler = lr_scheduler.StepLR(optimizer, step_size=7, gamma=0.1)
 
 model = train_model(model, criterion, optimizer, step_lr_scheduler, num_epochs=25)
 
