From 4e12daa7e1afa410089036e1a7426deda287288f Mon Sep 17 00:00:00 2001 From: Lazaros Toumanidis Date: Mon, 2 Sep 2019 00:22:06 +0300 Subject: [PATCH] Update transfer_learning_tutorial.py scheduler.step() per epoch not per batch --- beginner_source/transfer_learning_tutorial.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/beginner_source/transfer_learning_tutorial.py b/beginner_source/transfer_learning_tutorial.py index b6196149594..ef0ea69de4a 100644 --- a/beginner_source/transfer_learning_tutorial.py +++ b/beginner_source/transfer_learning_tutorial.py @@ -180,11 +180,12 @@ def train_model(model, criterion, optimizer, scheduler, num_epochs=25): if phase == 'train': loss.backward() optimizer.step() - scheduler.step() # statistics running_loss += loss.item() * inputs.size(0) running_corrects += torch.sum(preds == labels.data) + if phase == 'train': + scheduler.step() epoch_loss = running_loss / dataset_sizes[phase] epoch_acc = running_corrects.double() / dataset_sizes[phase]