config.json

{
  "channels": 5,
  "depths": [
    128,
    256,
    512,
    1024,
    1024
  ],
  "gn_groups": 16,
  "kernel_sizes": [
    7,
    5,
    3,
    3,
    3
  ],
  "num_layers": 5
}