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
}