config.json

{
  "model_type": "shadenet",
  "model_name": "ShadeNet-2-20M",
  "architecture": "ParallelUNet-v2",
  "params_total": 20050000,
  "params_trainable": 18240000,
  "image_size": 384,
  "in_ch": 3,
  "out_ch": 8,
  "framework": "pytorch"
}