Skip to content

Commit 1d32f4f

Browse files
Update beginner_source/knowledge_distillation_tutorial.py
Co-authored-by: Svetlana Karslioglu <svekars@meta.com>
1 parent 196637a commit 1d32f4f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

beginner_source/knowledge_distillation_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ def forward(self, x):
146146
# We employ 2 functions to help us produce and evaluate the results on our original classification task.
147147
# One function called train with the following arguments:
148148
#
149-
# - ``model``: A model instance that is to be trained (update its weights) via this function
149+
# - ``model``: A model instance to train (update its weights) via this function.
150150
# - ``train_loader``: We defined our train_loader above, its job is to feed the data into the model
151151
# - ``epochs``: How many times we loop over the dataset
152152
# - ``learning_rate``: How large do we want the steps to be towards convergence. Too large / small steps can be bad

0 commit comments

Comments
 (0)