config.json

{
  "d_model": 1,
  "dec_init_norm": 1,
  "hidden_activation": "topk",
  "hidden_activation_kwargs": {
    "k": 1
  },
  "hidden_dim": 1,
  "n_layers": 1,
  "n_models": 1
}