sgrvinod / a-PyTorch-Tutorial-to-Image-Captioning

Show, Attend, and Tell | a PyTorch Tutorial to Image Captioning
MIT License
2.77k stars 716 forks source link

How to do ensemble at testing with beam search? #89

Open homelifes opened 5 years ago

homelifes commented 5 years ago

Hi @sgrvinod, thanks for your code. When using beam search, how de we perform ensemble testing (testing multiple models and average predictions across models). Should we add all the log softmax output in the beam search fuction and then divide them by the number of models?

homelifes commented 5 years ago

@sgrvinod im waiting for you reply