Update model.py

2024-04-29 21:47:03 +02:00 · 2024-04-29 21:47:03 +02:00 · 6a0b357945
parent b45d036d42
commit 6a0b357945
1 changed files with 1 additions and 1 deletions
--- 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')