Zaktualizuj 'ml_pytroch_sacred.py'

This commit is contained in:
Sebastian Wałęsa 2022-05-09 11:27:48 +02:00
parent 06f03475f0
commit 94f58c9fb9

View File

@ -128,3 +128,4 @@ def my_main(epochs):
file.write(str(predict_single(input_, target, model))) file.write(str(predict_single(input_, target, model)))
torch.save(model, "Model_xPosition.pkl") torch.save(model, "Model_xPosition.pkl")
ex.add_artifact("Model_xPosition.pkl")