Spaces:

nssharmaofficial
/

ImageCaption

Sleeping

nssharmaofficial commited on Mar 30

Commit

4169569

•

1 Parent(s): 29003f1

Fix typo

Files changed (1) hide show

source/model.py CHANGED Viewed

@@ -84,7 +84,7 @@ class Decoder(nn.Module):
                             bidirectional=False)
         self.fc = nn.Sequential(
-            nn.Linear(in_features=self.hidden_dim, out_featuers=self.vocab_size),
             nn.LogSoftmax(dim=2)
         )

                             bidirectional=False)
         self.fc = nn.Sequential(
+            nn.Linear(in_features=self.hidden_dim, out_features=self.vocab_size),
             nn.LogSoftmax(dim=2)
         )