config.json

{
  "model_type": "unet",
  "dim": 200,
  "dim_mults": [1, 2, 4],
  "init_dim": null,
  "resnet_block_groups": 4,
  "num_classes": 10,
  "channels": 1
}