diff --git a/tutorials/01-basics/pytorch_basics/main.py b/tutorials/01-basics/pytorch_basics/main.py index 744400c2..03e83c5f 100644 --- a/tutorials/01-basics/pytorch_basics/main.py +++ b/tutorials/01-basics/pytorch_basics/main.py @@ -168,6 +168,8 @@ def __len__(self): param.requires_grad = False # Replace the top layer for finetuning. +# top layer (or fully connected layer) refers to +# the last layer before the output of the network resnet.fc = nn.Linear(resnet.fc.in_features, 100) # 100 is an example. # Forward pass.