From ffaabb46db8a8b2881a5950df448c03d74a008da Mon Sep 17 00:00:00 2001 From: Ben Shaver Date: Tue, 25 Dec 2018 12:52:40 -0500 Subject: [PATCH] Update transfer_learning_tutorial.py --- beginner_source/transfer_learning_tutorial.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/beginner_source/transfer_learning_tutorial.py b/beginner_source/transfer_learning_tutorial.py index 1d275b6e5d2..b236db687ca 100644 --- a/beginner_source/transfer_learning_tutorial.py +++ b/beginner_source/transfer_learning_tutorial.py @@ -305,7 +305,7 @@ def visualize_model(model, num_images=6): criterion = nn.CrossEntropyLoss() # Observe that only parameters of final layer are being optimized as -# opoosed to before. +# opposed to before. optimizer_conv = optim.SGD(model_conv.fc.parameters(), lr=0.001, momentum=0.9) # Decay LR by a factor of 0.1 every 7 epochs