From 182d136aca384501225ee8d91418a7763d72704f Mon Sep 17 00:00:00 2001 From: Adrian Charkiewicz Date: Sun, 15 May 2022 21:33:15 +0200 Subject: [PATCH] typo in signature --- pytorch/pytorch.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pytorch/pytorch.py b/pytorch/pytorch.py index e54a5f1..f06db1e 100644 --- a/pytorch/pytorch.py +++ b/pytorch/pytorch.py @@ -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)