config.json
| 1 | { |
| 2 | "vocab_size": 8000, |
| 3 | "d_model": 256, |
| 4 | "nhead": 4, |
| 5 | "num_layers": 4, |
| 6 | "dim_ff": 1024, |
| 7 | "dropout": 0.3, |
| 8 | "max_len": 128, |
| 9 | "pad": 0, |
| 10 | "unk": 1, |
| 11 | "bos": 2, |
| 12 | "eos": 3 |
| 13 | } |