a-PyTorch-Tutorial-to-Image-Captioning
a-PyTorch-Tutorial-to-Image-Captioning copied to clipboard
Loading model back to generate caption meet ERROR
/envs/caption/lib/python3.7/site-packages/torch/nn/modules/module.py", line 539, in __getattr__ type(self).__name__, name)) AttributeError: 'DecoderWithAttention' object has no attribute 'decode_step'
Hello, Thanks for this good repo .
When I got my model, I load it back to generate the caption. There is an ERROR, like above showing.
seems like something wrong with the dumping or loading period ?