config.json

{
  "input_size": 512,
  "output_size": 2
}