From e5ade0a1669a4555f978a2013c4b009ec0e8b306 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebastian=20Wa=C5=82=C4=99sa?= Date: Sun, 15 May 2022 22:06:15 +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 6671557..37979a2 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))) - print(val_ds) + print(val_ds[0]) # input_example = val_ds[0] # # input_example = input_example.unsqueeze(0)