config.json

{
  "batch_size": 16,
  "criterion": "mse",
  "grayScale": true,
  "img_size": 224,
  "lr": 0.004,
  "max_epochs": 50,
  "optimizer": "adam",
  "out_dim": 136,
  "patience": 10,
  "pretrained_backbone": true,
  "random_seed": 42,
  "use_scheduler": true
}