From acc735d7ba6aa10e4d4192f25ce2d259a05e8a39 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebastian=20Wa=C5=82=C4=99sa?= Date: Sun, 15 May 2022 22:00:38 +0200 Subject: [PATCH] Zaktualizuj 'ml_pytorch_mlflow.py' --- ml_pytorch_mlflow.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/ml_pytorch_mlflow.py b/ml_pytorch_mlflow.py index dfb1605..9239cbd 100644 --- a/ml_pytorch_mlflow.py +++ b/ml_pytorch_mlflow.py @@ -146,11 +146,13 @@ def my_main(epochs): for i in range(0, len(val_ds), 1): input_, target = val_ds[i] file.write(str(predict_single(input_, target, model))) - - input_example = val_ds[0] - # input_example = input_example.unsqueeze(0) - signature = infer_signature(input_[0], prediction(input_[0], model)) - tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme + + print(input_) + + # input_example = val_ds[0] + # # input_example = input_example.unsqueeze(0) + # signature = infer_signature(input_[0], prediction(input_[0], model)) + # tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme if tracking_url_type_store != "file": mlflow.pytorch.log_model(model, "model", registered_model_name="s478839", signature=siganture,