| { | |
| "d_model": 1024, | |
| "nhead": 4, | |
| "num_layers": 2, | |
| "dim_feedforward": 2048, | |
| "out_dim": 512, | |
| "dropout": 0.1, | |
| "activation": "relu", | |
| "lr0": 0.0001, | |
| "warmup_steps": 300 | |
| } |
| { | |
| "d_model": 1024, | |
| "nhead": 4, | |
| "num_layers": 2, | |
| "dim_feedforward": 2048, | |
| "out_dim": 512, | |
| "dropout": 0.1, | |
| "activation": "relu", | |
| "lr0": 0.0001, | |
| "warmup_steps": 300 | |
| } |