Merge pull request huggingface#9 from gyes00205/master

Update 16_tensorboard.py

Author: patrickloeber

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