Merge pull request clovaai#181 from tjdevWorks/master

Doc String Correction: Attention class forward function

modules/prediction.py

@@ -23,7 +23,7 @@ def _char_to_onehot(self, input_char, onehot_dim=38):
     def forward(self, batch_H, text, is_train=True, batch_max_length=25):
         """
         input:
-            batch_H : contextual_feature H = hidden state of encoder. [batch_size x num_steps x num_classes]
+            batch_H : contextual_feature H = hidden state of encoder. [batch_size x num_steps x contextual_feature_channels]
             text : the text-index of each image. [batch_size x (max_length+1)]. +1 for [GO] token. text[:, 0] = [GO].
         output: probability distribution at each step [batch_size x num_steps x num_classes]
         """