Initialize DecoderCNN in Image captioning

yunjey / pytorch-tutorial

PyTorch Tutorial for Deep Learning Researchers

MIT License

30.36k stars 8.14k forks source link

Initialize DecoderCNN in Image captioning #218

Open karan469 opened 4 years ago

karan469 commented 4 years ago

The performance of the model with those default hyper parameters take larger time to train and unusually larger loss at starting point. Can you please tell how to initialize the DecoderCNN using Xavier initialization?