File size: 194 Bytes
666fa9d |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
{
"params": {
"arch": "FPN",
"encoder_name": "resnet34",
"in_channels": 3,
"out_classes": 1
},
"mpp": 10,
"size": 2048,
"train_batch_size": 4,
"val_batch_size": 16,
"epochs": 10
} |