From 96f600983f9b6daf0e2e619abbce1e1d548350f3 Mon Sep 17 00:00:00 2001
From: Uzaaft <48220549+Uzaaft@users.noreply.github.com>
Date: Fri, 17 Jan 2020 12:58:53 +0100
Subject: [PATCH] Update train.py

Changed line 98 to fix Issue nr #12
---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index efe631d..08b75df 100644
--- a/train.py
+++ b/train.py
@@ -95,7 +95,7 @@
         #Log to training loss/acc
         print('Iteration:', i, 'Training loss:', var_to_np(loss))
         if log_tensorboard:
-            logger.log_scalar('training_loss',  var_to_np(loss),i)
+            logger.log_scalar('training_loss',  var_to_np(loss).tolist(),i)
         for k,v in computeAccuracy(torch.argmax(output,1), labels).items():
             if log_tensorboard:
                 logger.log_scalar('training_' + k, v, i)