{
  "architecture": "UNet",
  "encoder": "resnet34",
  "encoder_weights": "imagenet",
  "in_channels": 3,
  "classes": 1,
  "input_size": 256,
  "best_dice_score": 0.9559824578464031,
  "best_iou_score": 0.9162851329892874,
  "epochs_trained": 10,
  "batch_size": 16,
  "learning_rate": 0.0001,
  "dataset": "mattmdjaga/human_parsing_dataset",
  "loss_function": "DiceLoss + BCE",
  "optimizer": "Adam"
}