fixed running accuracy calculation in 16_tensorboard

patrickloeber · patrickloeber · commit e6dd99c0ffed · 2020-10-13T17:38:45.000+02:00
diff --git a/16_tensorboard.py b/16_tensorboard.py
@@ -113,7 +113,8 @@ def forward(self, x):
             print (f'Epoch [{epoch+1}/{num_epochs}], Step [{i+1}/{n_total_steps}], Loss: {loss.item():.4f}')
             ############## TENSORBOARD ########################
             writer.add_scalar('training loss', running_loss / 100, epoch * n_total_steps + i)
-            writer.add_scalar('accuracy', running_correct / 100, epoch * n_total_steps + i)
+            running_accuracy = running_correct / 100 / predicted.size(0)
+            writer.add_scalar('accuracy', running_accuracy, epoch * n_total_steps + i)
             running_correct = 0
             running_loss = 0.0
             ###################################################
