.

training.py

@@ -64,8 +64,8 @@ model.compile(
 # Train model
 history = model.fit(
     X_train, y_train,
-    batch_size=sys.argv[0],
-    epochs=sys.argv[1],
+    batch_size=int(sys.argv[0]),
+    epochs=int(sys.argv[1]),
     validation_split=0.2)
 
 test_results = {}