From 5687a9cc9870b5df737b4e6fe9e37d8e92691a69 Mon Sep 17 00:00:00 2001
From: Maciej Sobkowiak <masobkowiak@gmail.com>
Date: Sun, 16 May 2021 16:55:33 +0200
Subject: [PATCH] .

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

diff --git a/training.py b/training.py
index d01674a..0348e6b 100644
--- a/training.py
+++ b/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 = {}