config.json

{
  "input_dim": 768,
  "hidden_size": 768,
  "output_dim": 2
}