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)