Zaktualizuj 'ml_pytorch_mlflow.py'

This commit is contained in:
Sebastian Wałęsa 2022-05-15 23:04:18 +02:00
parent 22b7a65a17
commit 216a56f040

View File

@ -137,7 +137,7 @@ def my_main(epochs):
input_, target = val_ds[i] input_, target = val_ds[i]
expected.append(float(target)) expected.append(float(target))
predicted.append(float(prediction(input_, model))) predicted.append(float(prediction(input_, model)))
inputss.append(input_) # inputss.append(input_)
inputss, targetss = val_ds inputss, targetss = val_ds