From 6a0b357945b0a757b1148c6fedb7d9ab7d2a293c Mon Sep 17 00:00:00 2001
From: Alicja Szulecka <73056579+AliSzu@users.noreply.github.com>
Date: Mon, 29 Apr 2024 21:47:03 +0200
Subject: [PATCH] Update model.py

---
 model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model.py b/model.py
index 12e8a95..4bd3e73 100644
--- a/model.py
+++ b/model.py
@@ -31,7 +31,7 @@ class Model(nn.Module):
         return x
 
 def main():
-    epochs = sys.argv[1]
+    epochs = int(sys.argv[1])
     print(epochs)
 
     forest_train = pd.read_csv('forest_train.csv')