From 0728bec2c0b79f2d4a0f2f24421e33aa73eb959d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebastian=20Wa=C5=82=C4=99sa?= Date: Sun, 15 May 2022 14:20:17 +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 80e559d..bf06f9e 100644 --- a/ml_pytorch_mlflow.py +++ b/ml_pytorch_mlflow.py @@ -147,7 +147,7 @@ def my_main(epochs): input_, target = val_ds[i] file.write(str(predict_single(input_, target, model))) - input_example = val_ds[0].unsqueeze(0) + input_example = val_ds[0] signature = mlflow.models.signature.infer_signature(input_, prediction(input_, model)) tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme