| { | |
| "model_name": "Seq2Seq", | |
| "task": "translation", | |
| "framework": "PyTorch", | |
| "dataset": "bentrevett/multi30k", | |
| "src_lang": "en", | |
| "tgt_lang": "de", | |
| "src_vocab_size": 5968, | |
| "tgt_vocab_size": 7812, | |
| "architecture": { | |
| "type": "Seq2Seq", | |
| "cell": "GRU", | |
| "emb_dim": 128, | |
| "hidden_dim": 256 | |
| }, | |
| "training": { | |
| "optimizer": "Adam", | |
| "learning_rate": 0.001, | |
| "loss": "CrossEntropyLoss", | |
| "grad_clip_norm": 1.0, | |
| "epochs": 50, | |
| "batch_size": 64 | |
| } | |
| } |