SpongeBoSquarePants

> When reproducing the code, using the "Medt" network, you have not encountered the training and test results of the situation is very different Me too.I met the same issue...

The issue is disappear after I delete model.eval() in the test.py file.I don't know why.