Fix for training

2021-05-15 09:06:59 +02:00 · 2021-05-15 09:06:59 +02:00 · e3aaf3d720
commit e3aaf3d720
parent 8929851757
1 changed files with 2 additions and 2 deletions
--- 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)