Futher fix the sample_beam in CaptionModel.

Author: ruotianluo

models/CaptionModel.py

@@ -95,9 +95,10 @@ def sample_beam(self, fc_feats, att_feats, opt={}):
 
         self.done_beams = [[] for _ in range(batch_size)]
         for k in range(batch_size):
-            state = self.init_hidden(fc_feats[k:k+1]).expand(beam_size, self.rnn_size)
             tmp_fc_feats = fc_feats[k:k+1].expand(beam_size, self.fc_feat_size)
             tmp_att_feats = att_feats[k:k+1].expand(*((beam_size,)+att_feats.size()[1:]))
+            
+            state = self.init_hidden(tmp_fc_feats)
 
             beam_seq = torch.LongTensor(self.seq_length, beam_size).zero_()
             beam_seq_logprobs = torch.FloatTensor(self.seq_length, beam_size).zero_()