Skip to content

Commit d166fff

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

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
@@ -88,7 +88,7 @@
8888
# Both architectures are Convolutional Neural Networks (CNNs) with a different number of convolutional layers that serve as feature extractors, followed by a classifier with 10 classes.
8989
# The number of filters and neurons is smaller for the students.
9090

91-
# Deeper neural network class to be used as teacher
91+
# Deeper neural network class to be used as teacher:
9292
class DeepNN(nn.Module):
9393
def __init__(self, num_classes=10):
9494
super(DeepNN, self).__init__()

0 commit comments

Comments
 (0)