Update model.py

This commit is contained in:
Alicja Szulecka 2024-04-29 21:47:03 +02:00
parent b45d036d42
commit 6a0b357945

View File

@ -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')