Merge pull request huggingface#9 from gyes00205/master

Update 16_tensorboard.py
lcong · Jan 5, 2023 · 75d768b · 75d768b
2 parents e6dd99c + 43697b6
commit 75d768b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/16_tensorboard.py b/16_tensorboard.py
@@ -138,7 +138,7 @@ def forward(self, x):
         class_probs_batch = [F.softmax(output, dim=0) for output in outputs]
 
         class_preds.append(class_probs_batch)
-        class_labels.append(predicted)
+        class_labels.append(labels)
 
     # 10000, 10, and 10000, 1
     # stack concatenates tensors along a new dimension