Zaktualizuj 'train-sacred.py'

This commit is contained in:
Kornelia Girejko 2022-05-08 19:18:50 +02:00
parent fd7bc6bd45
commit 8cfe58ca38

View File

@ -96,7 +96,7 @@ def my_main(epochs):
result = open("output",'w+')
result.write(f'{y_predicted}')
torch.save(model, "modelP.pkl")
torch.save(model, "modelS.pkl")
ex.run()