Omnia-cy commited on
Commit
72759f6
·
verified ·
1 Parent(s): 7dafdfc

Upload 3 files

Browse files
Files changed (2) hide show
  1. best_model.pt +3 -0
  2. config.json +1 -18
best_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0ea19eb8271d2ef53ff5d3a02b3616d60235c266026e4b965ac4f04f1392a914
3
+ size 30825300
config.json CHANGED
@@ -1,18 +1 @@
1
- import json
2
-
3
- config = {
4
- "src_vocab_size": SourceVocab,
5
- "tgt_vocab_size": TargetVocab,
6
- "max_src_len": MaxSrcLength,
7
- "max_tgt_len": MaxTgtLength,
8
- "pad_id": 0,
9
- "bos_id": 2,
10
- "eos_id": 3,
11
- "d_model": 256,
12
- "num_heads": 4,
13
- "num_layers": 3,
14
- "d_ff": 512
15
- }
16
-
17
- with open("config.json", "w") as f:
18
- json.dump(config, f)
 
1
+ {"src_vocab_size": 2500, "tgt_vocab_size": 6000, "max_src_len": 85, "max_tgt_len": 87, "pad_id": 0, "bos_id": 2, "eos_id": 3, "d_model": 256, "num_heads": 4, "num_layers": 3, "d_ff": 512}