From 12e394819dd502b445e39691cf156405f9f4d965 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Wed, 15 Jul 2020 15:29:26 +0000
Subject: [PATCH] Fixed output of train method
---
main/train.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main/train.py b/main/train.py
index 9843012..a14e22a 100755
--- a/main/train.py
+++ b/main/train.py
@@ -27,4 +27,4 @@ if __name__ == '__main__':
# train the network
print('----------------------- Network training -------------------------')
- loss, accuracy, vloss, vaccuracy = trainer.train()
+ training_state = trainer.train()
--
GitLab