Update beginner_source/knowledge_distillation_tutorial.py

AlexandrosChrtn · svekars · web-flow · commit 196637a7d9b2 · 2023-08-16T23:17:49.000+03:00
Co-authored-by: Svetlana Karslioglu &lt;svekars@meta.com&gt;
diff --git a/beginner_source/knowledge_distillation_tutorial.py b/beginner_source/knowledge_distillation_tutorial.py
@@ -150,7 +150,7 @@ def forward(self, x):
 # - ``train_loader``: We defined our train_loader above, its job is to feed the data into the model
 # - ``epochs``: How many times we loop over the dataset
 # - ``learning_rate``: How large do we want the steps to be towards convergence. Too large / small steps can be bad
-# - ``device``: depends on CPU or GPU availability
+# - ``device``: Determines the device to run the workload on. Can be either CPU or GPU depending on availability.
 #
 # Our test function is similar, but it will be invoked with our ``test_loader``, to load images from our test set.
 #

Original file line number	Diff line number	Diff line change
`@@ -150,7 +150,7 @@ def forward(self, x):`
`150`	`150`	# - ``train_loader``: We defined our train_loader above, its job is to feed the data into the model
`151`	`151`	# - ``epochs``: How many times we loop over the dataset
`152`	`152`	# - ``learning_rate``: How large do we want the steps to be towards convergence. Too large / small steps can be bad
`153`		-# - ``device``: depends on CPU or GPU availability
	`153`	+# - ``device``: Determines the device to run the workload on. Can be either CPU or GPU depending on availability.
`154`	`154`	`#`
`155`	`155`	# Our test function is similar, but it will be invoked with our ``test_loader``, to load images from our test set.
`156`	`156`	`#`