config.json

{
  "dims": [
    128,
    320,
    512
  ],
  "dropout": 0.1,
  "expand_ratios": [
    4,
    4,
    2,
    2
  ],
  "init_dim": 64,
  "layer_depths": [
    3,
    3,
    12,
    2
  ],
  "num_classes": 1000,
  "num_layer": 4
}