pytorch-tutorial icon indicating copy to clipboard operation
pytorch-tutorial copied to clipboard

Can transfer learning be performed from the pre-trained image caption model

Open kevjp opened this issue 6 years ago • 1 comments

From image captioning tutorial is there any way to continue training from the pre-trained model? If so how do we load the model's state_dict and the optimizer's state_dict? As far as i can make out I am unable to load these attributes from the pickle files decoder-5-3000.pkl and encoder-5-3000.pkl files.

So I can't perform the following: checkpoint = torch.load('decoder-5-3000.pkl') model.load_state_dict(checkpoint['model_state_dict']) optimizer.load_state_dict(checkpoint['optimizer_state_dict'])

Any help would be much appreciated.

kevjp avatar Sep 16 '19 13:09 kevjp

good afternoon, can you share the pretrained.zip? Thanks a lot

wys837 avatar Jul 07 '20 08:07 wys837