Again further fix bugs in CaptionModel in sample_beam

Author: ruotianluo

models/CaptionModel.py

@@ -96,7 +96,7 @@ def sample_beam(self, fc_feats, att_feats, opt={}):
         self.done_beams = [[] for _ in range(batch_size)]
         for k in range(batch_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:]))
+            tmp_att_feats = att_feats[k:k+1].expand(*((beam_size,)+att_feats.size()[1:])).contiguous()
 
             state = self.init_hidden(tmp_fc_feats)