s444354/ium_444354
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

typo in signature
Some checks failed
s444354-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Adrian Charkiewicz 2022-05-15 21:33:15 +02:00
parent 3d32874c63
commit 182d136aca
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pytorch/pytorch.py
View File

@ -194,10 +194,10 @@ def fit(epochs, lr, model, train_loader, val_loader, _log, _run, opt_func=torch.
input_example = [0.1, 0.1, 546.0, 555.223, 1., 1., 33.16376, 84.12426]
if tracking_url_type_store != "file":
mlflow.pytorch.log_model(model, "model", registered_model_name="s444354", signature=siganture, input_example=input_example)
mlflow.pytorch.log_model(model, "model", registered_model_name="s444354", signature=signature, input_example=input_example)
else:
mlflow.pytorch.log_model(model, "model", signature=siganture, input_example=input_example)
mlflow.pytorch.save_model(model, "my_model", signature=siganture, input_example=input_example)
mlflow.pytorch.log_model(model, "model", signature=signature, input_example=input_example)
mlflow.pytorch.save_model(model, "my_model", signature=signature, input_example=input_example)