From e3aaf3d720bf587b9c64d1cb7330ae2e1bac6269 Mon Sep 17 00:00:00 2001
From: Wojciech Jarmosz <wojciech@jarmosz.com>
Date: Sat, 15 May 2021 09:06:59 +0200
Subject: [PATCH] Fix for training

---
 training.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/training.py b/training.py
index 68e91fe..98c7089 100644
--- a/training.py
+++ b/training.py
@@ -11,8 +11,8 @@ from tensorflow.keras.layers.experimental import preprocessing
 
 arguments = sys.argv[1:]
 
-verbose = arguments[0]
-epochs = arguments[1]
+verbose = int(arguments[0])
+epochs = int(arguments[1])
 
 pd.set_option("display.max_columns", None)