test_custom_rnn / config.json
cartesinus's picture
Add custom RNN model with attention
2d37aac verified
raw
history blame contribute delete
163 Bytes
{"model_type": "custom_rnn", "vocab_size": 250002, "hidden_size": 256, "output_size": 2, "cell_type": "RNN", "architecture": "SimpleRecurrentNetworkWithAttention"}