Missing switch back to train() mode

[iamgroot42]

Hello. In the following lines:

FLIP/invert_CIFAR.py

Line 85 in b3fd729

self.model.eval()

and

FLIP/invert_CIFAR.py

Line 321 in b3fd729

self.model.eval()

The model is set to eval() mode (thus batch-norm statistics are not tracked) but is never reverted to train() as it should be, since the model is still in the process of training.

[KaiyuanZh]

Hi @iamgroot42, thank you for reporting this. I have fixed it. Please let me know if you encounter any other issues.