Fix small error in display

epfl-lara · Mar 27, 2018 · ee47423 · ee47423
1 parent ba0f959
commit ee47423
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nugget/train.py b/nugget/train.py
@@ -177,7 +177,7 @@
             sys.stdout.write("\033[F")
             sys.stdout.write("\033[F")
             sys.stdout.write("\033[K")
-            print("{0:{1}d} / {2}".format(n, len(str(total)), total))
+            print("{0:{1}d} / {2}".format(n, len(str(val_total)), val_total))
             sys.stdout.write("\033[K")
             print("MAE for distances: {}".format(absolute_error_distance / n))
             sys.stdout.write("\033[K")