config.json

{
  "img_channel": 3,
  "width": 64,
  "middle_blk_num": 1,
  "enc_blk_nums": [
    1,
    1,
    1,
    28
  ],
  "dec_blk_nums": [
    1,
    1,
    1,
    1
  ],
  "local": true,
  "train_size": [
    1,
    3,
    256,
    256
  ]
}