Covert pth model error

[shuguohe]

@ruotianluo I tried to convert model-best.pth to tensorflow .pb model, but got the error, File "/home/ubuntu/image_caption/self-critical.pytorch/models/CaptionModel.py", line 33, in forward return getattr(self, '_'+mode)(*args, **kwargs) TypeError: _forward() takes at least 4 arguments (2 given) In Attmodel.py, def _forward(self, fc_feats, att_feats, seq, att_masks=None): so the parameter count is not match.