From 312a7b3bb334a3d3d59ca3da26fcc1ee6593a701 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebastian=20Wa=C5=82=C4=99sa?= Date: Sun, 15 May 2022 14:24:04 +0200 Subject: [PATCH] Zaktualizuj 'ml_pytorch_mlflow.py' --- ml_pytorch_mlflow.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ml_pytorch_mlflow.py b/ml_pytorch_mlflow.py index 29eef9b..38f808a 100644 --- a/ml_pytorch_mlflow.py +++ b/ml_pytorch_mlflow.py @@ -148,7 +148,7 @@ def my_main(epochs): file.write(str(predict_single(input_, target, model))) input_example = val_ds[0] - input_example = input_example.unsqueeze(0) + # input_example = input_example.unsqueeze(0) signature = mlflow.models.signature.infer_signature(input_, prediction(input_example, model)) tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme