| { | |
| "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" | |
| } |