cannot evaluate after training

[chongtianyifa]

Hi,

I successfully train the model using a file. However, it did not allow me to evaluate the model by call vame.evaluate_model(config).

I attach my model and data.
Please let me if you need other files.

Thank you
HFpress-Nov5-2022.zip

[K L]

Hey! I figured that this is a problem due to a newer version of pytorch.

You would need to replace x = test_loader.__iter__().next() by the two following lines of code:

x_iter = iter(test_loader)
x = next(x_iter)

I hope this quick fix will work for you.