-
-
Notifications
You must be signed in to change notification settings - Fork 17
/
2020-11-08.yaml
137 lines (124 loc) · 2.95 KB
/
2020-11-08.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
---
seed: 1984
num_workers: 4
experiment_name: "2020-11-08"
model:
type: timm.create_model
model_name: swsl_resnext50_32x4d
num_classes: 4
pretrained: True
trainer:
type: pytorch_lightning.Trainer
gpus: 4
max_epochs: 30
distributed_backend: ddp
progress_bar_refresh_rate: 1
benchmark: True
precision: 16
gradient_clip_val: 5.0
num_sanity_val_steps: 2
sync_batchnorm: True
scheduler:
type: torch.optim.lr_scheduler.CosineAnnealingWarmRestarts
T_0: 10
T_mult: 2
train_parameters:
batch_size: 64
checkpoint_callback:
type: pytorch_lightning.callbacks.ModelCheckpoint
filepath: "2020-11-08"
monitor: val_loss
verbose: True
mode: min
save_top_k: -1
val_parameters:
batch_size: 64
optimizer:
type: torch.optim.SGD
momentum: 0.9
lr: 0.1
loss:
type: torch.nn.CrossEntropyLoss
train_aug:
transform:
__class_fullname__: albumentations.core.composition.Compose
bbox_params: null
keypoint_params: null
p: 1
transforms:
- __class_fullname__: albumentations.augmentations.transforms.PadIfNeeded
always_apply: False
min_height: 224
min_width: 224
border_mode: 0 # cv2.BORDER_CONSTANT
value: 0
mask_value: 0
p: 1
- __class_fullname__: albumentations.augmentations.transforms.RandomCrop
always_apply: False
height: 224
width: 224
p: 1
- __class_fullname__: albumentations.augmentations.transforms.Normalize
always_apply: false
max_pixel_value: 255.0
mean:
- 0.485
- 0.456
- 0.406
p: 1
std:
- 0.229
- 0.224
- 0.225
val_aug:
transform:
__class_fullname__: albumentations.core.composition.Compose
bbox_params: null
keypoint_params: null
p: 1
transforms:
- __class_fullname__: albumentations.augmentations.transforms.Resize
always_apply: False
height: 224
width: 224
p: 1
- __class_fullname__: albumentations.augmentations.transforms.HorizontalFlip
always_apply: False
p: 0.5
- __class_fullname__: albumentations.augmentations.transforms.Normalize
always_apply: false
max_pixel_value: 255.0
mean:
- 0.485
- 0.456
- 0.406
p: 1
std:
- 0.229
- 0.224
- 0.225
test_aug:
transform:
__class_fullname__: albumentations.core.composition.Compose
bbox_params: null
keypoint_params: null
p: 1
transforms:
- __class_fullname__: albumentations.augmentations.transforms.Resize
always_apply: False
height: 224
width: 224
p: 1
- __class_fullname__: albumentations.augmentations.transforms.Normalize
always_apply: false
max_pixel_value: 255.0
mean:
- 0.485
- 0.456
- 0.406
p: 1
std:
- 0.229
- 0.224
- 0.225