Hi jemery.  The reason why learn.module(input_img) can not predict as well as learn.prrdict was because I didn’t load the opt state dict? Or after I set eval() and require grad is false just like lesson7 notebook, learn.module(input_img) will predict as well as learn.predict? Thank you