Spaces:

nssharmaofficial
/

ImageCaption

Sleeping

nssharmaofficial commited on 23 days ago

Commit

81de229

•

1 Parent(s): 9f7e6ed

Fix typo in config

Files changed (1) hide show

source/config.py CHANGED Viewed

@@ -13,7 +13,7 @@ class Config(object):
         self.NUM_LAYER = 1
         self.IMAGE_EMB_DIM = 512
-        self.WORD_EMB_DIM = 5121
         self.HIDDEN_DIM = 512
         self.EMBEDDING_WEIGHT_FILE = 'source/weights/embeddings-32B-512H-1L-e10.pt'

         self.NUM_LAYER = 1
         self.IMAGE_EMB_DIM = 512
+        self.WORD_EMB_DIM = 512
         self.HIDDEN_DIM = 512
         self.EMBEDDING_WEIGHT_FILE = 'source/weights/embeddings-32B-512H-1L-e10.pt'