Zaktualizuj 'train-sacred.py'
This commit is contained in:
parent
fd7bc6bd45
commit
8cfe58ca38
@ -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()
|
Loading…
Reference in New Issue
Block a user