Spaces:
Sleeping
Sleeping
Upload 3 files
Browse files- best_model.pt +3 -0
- 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 |
-
|
| 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}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|