nssharmaofficial commited on
Commit
9f7e6ed
1 Parent(s): 9a90e40

Fix weights path

Browse files
Files changed (1) hide show
  1. source/config.py +3 -3
source/config.py CHANGED
@@ -16,7 +16,7 @@ class Config(object):
16
  self.WORD_EMB_DIM = 5121
17
  self.HIDDEN_DIM = 512
18
 
19
- self.EMBEDDING_WEIGHT_FILE = 'source/weights/embeddings-32B-512H-1L-e5.pt'
20
- self.ENCODER_WEIGHT_FILE = 'source/weights/encoder-32B-512H-1L-e5.pt'
21
- self.DECODER_WEIGHT_FILE = 'source/weights/decoder-32B-512H-1L-e5.pt'
22
 
 
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'
20
+ self.ENCODER_WEIGHT_FILE = 'source/weights/encoder-32B-512H-1L-e10.pt'
21
+ self.DECODER_WEIGHT_FILE = 'source/weights/decoder-32B-512H-1L-e10.pt'
22