config.json

{
  "input_dim": 2,
  "output_dim": 3,
  "global_feature_dim": 1024
}