nssharmaofficial commited on
Commit
d471ee2
·
verified ·
1 Parent(s): f83495d

Config with new models

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 = 256
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 = 256
17
  self.HIDDEN_DIM = 512
18
 
19
+ self.EMBEDDING_WEIGHT_FILE = 'source/weights/embeddings-32B-512H-1L-e2.pt'
20
+ self.ENCODER_WEIGHT_FILE = 'source/weights/encoder-32B-512H-1L-e2.pt'
21
+ self.DECODER_WEIGHT_FILE = 'source/weights/decoder-32B-512H-1L-e2.pt'
22