From 4be75b3877c698ff9f2a5e6c586fd3c243bedf4a Mon Sep 17 00:00:00 2001 From: Tejas Mahajan Date: Mon, 15 Jun 2020 22:22:51 +0530 Subject: [PATCH] Doc String Correction: Attention class forward function --- modules/prediction.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/prediction.py b/modules/prediction.py index c8a40af..b6c3cb3 100755 --- a/modules/prediction.py +++ b/modules/prediction.py @@ -23,7 +23,7 @@ class Attention(nn.Module): 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] """