Zaktualizuj 'ml_pytorch_mlflow.py'

This commit is contained in:
Sebastian Wałęsa 2022-05-15 23:08:32 +02:00
parent 74c2429fa7
commit cc3a00146c

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[i], targetss[i] = val_ds[i] inputss.append(input_)
# inputss, targetss = val_ds # inputss, targetss = val_ds