how to get the output image after model()

[wyhimself]

I have successfully run the command test, and also run another command train. When I read this code, i found that there results = self.module.step(data, train=True, epoch=epoch, step=gstep) didn't return item 'imgs', just calculate with tensor's multi-dimensional array.

[wyhimself]

so how can I get the img? I found that there's func for save_img, is it implement or not?