ImageCaptioning.pytorch
ImageCaptioning.pytorch copied to clipboard
About using cpu on evaluating
Hi~ Thanks for sharing the codes. I have trained a gpu model using my own dataset. It really helps a lot. However, now I need evaluate the model on another machine only with cpu. So could you help provide some codes about how to convert gpu model to a cpu checkpoint, and how to eval using cpu model? Thanks a lot!