Hugging Face's logo

Spaces:
nssharmaofficial
/
ImageCaption
Sleeping

App Files Community
nssharmaofficial commited on
Commit
81de229
•
1 Parent(s): 9f7e6ed

Fix typo in config

Browse files
Files changed (1) hide show
  1. source/config.py +1 -1
source/config.py CHANGED
@@ -13,7 +13,7 @@ class Config(object):
13
 
14
  self.NUM_LAYER = 1
15
  self.IMAGE_EMB_DIM = 512
16
- self.WORD_EMB_DIM = 5121
17
  self.HIDDEN_DIM = 512
18
 
19
  self.EMBEDDING_WEIGHT_FILE = 'source/weights/embeddings-32B-512H-1L-e10.pt'
 
13
 
14
  self.NUM_LAYER = 1
15
  self.IMAGE_EMB_DIM = 512
16
+ self.WORD_EMB_DIM = 512
17
  self.HIDDEN_DIM = 512
18
 
19
  self.EMBEDDING_WEIGHT_FILE = 'source/weights/embeddings-32B-512H-1L-e10.pt'