spandan21 commited on
Commit
c2d8c4a
1 Parent(s): 0a725a9

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +42 -0
config.json ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": ["Unet"],
3
+ "model_type": "unet",
4
+ "backbone": "vgg19",
5
+ "input_channels": 3,
6
+ "num_classes": 3,
7
+ "epochs": 10,
8
+ "lr": 0.0002,
9
+ "scheduler": "CosineAnnealingLR",
10
+ "min_lr": 0.000001,
11
+ "T_max": 300,
12
+ "T_0": 25,
13
+ "warmup_epochs": 0,
14
+ "weight_decay": 0.000001,
15
+ "batch_size": 64,
16
+ "image_size": [224, 224],
17
+ "augmentation_strategy": {
18
+ "Resize": [224, 224],
19
+ "HorizontalFlip": 0.5,
20
+ "ShiftScaleRotate": {
21
+ "shift_limit": 0.0625,
22
+ "scale_limit": 0.05,
23
+ "rotate_limit": 10
24
+ },
25
+ "GridDistortion": {
26
+ "num_steps": 5,
27
+ "distort_limit": 0.05
28
+ },
29
+ "ElasticTransform": {
30
+ "alpha": 1,
31
+ "sigma": 50,
32
+ "alpha_affine": 50
33
+ },
34
+ "CoarseDropout": {
35
+ "max_holes": 8,
36
+ "max_height": 28,
37
+ "max_width": 28,
38
+ "min_holes": 5,
39
+ "fill_value": 0
40
+ }
41
+ }
42
+ }