| { | |
| "embedding_dim": 4, | |
| "hidden_size": 8, | |
| "num_layers": 1, | |
| "vocab_size_config": 50, | |
| "seq_length": 10, | |
| "use_gru": true, | |
| "rnn_activation_function": "tanh", | |
| "_commit_hash": null, | |
| "architectures": [ | |
| "WordRNNModel" | |
| ], | |
| "model_type": "gru", | |
| "bos_token_id": null, | |
| "eos_token_id": null, | |
| "pad_token_id": null, | |
| "unk_token_id": 49, | |
| "torch_dtype": "torch.float32", | |
| "vocab_size": 50 | |
| } |