File tree Expand file tree Collapse file tree 3 files changed +92
-2
lines changed
src/otx/recipe/classification Expand file tree Collapse file tree 3 files changed +92
-2
lines changed Original file line number Diff line number Diff line change
1
+ model :
2
+ class_path : otx.algo.classification.hlabel_models.timm_model.TimmModelHLabelCls
3
+ init_args :
4
+ model_name : mobilenetv4_medium_conv
5
+
6
+ optimizer :
7
+ class_path : torch.optim.SGD
8
+ init_args :
9
+ lr : 0.0058
10
+ momentum : 0.9
11
+ weight_decay : 0.0001
12
+
13
+ scheduler :
14
+ class_path : otx.core.schedulers.LinearWarmupSchedulerCallable
15
+ init_args :
16
+ num_warmup_steps : 10
17
+ main_scheduler_callable :
18
+ class_path : lightning.pytorch.cli.ReduceLROnPlateau
19
+ init_args :
20
+ mode : max
21
+ factor : 0.5
22
+ patience : 3
23
+ monitor : val/accuracy
24
+
25
+ engine :
26
+ task : H_LABEL_CLS
27
+ device : auto
28
+
29
+ callback_monitor : val/accuracy
30
+
31
+ data : ../../_base_/data/classification.yaml
32
+ overrides :
33
+ max_epochs : 90
34
+
35
+ callbacks :
36
+ - class_path : otx.algo.callbacks.adaptive_early_stopping.EarlyStoppingWithWarmup
37
+ init_args :
38
+ patience : 5
39
+
40
+ data :
41
+ task : H_LABEL_CLS
42
+ data_format : datumaro
Original file line number Diff line number Diff line change 1
1
model :
2
2
class_path : otx.algo.classification.multiclass_models.timm_model.TimmModelMulticlassCls
3
3
init_args :
4
- model_name : mobilenetv4_conv_small
4
+ model_name : mobilenetv4_medium_conv
5
5
label_info : 1000
6
+ freeze_backbone : False
7
+
6
8
optimizer :
7
9
class_path : torch.optim.SGD
8
10
init_args :
9
11
lr : 0.0058
10
12
momentum : 0.9
11
13
weight_decay : 0.0001
14
+
12
15
scheduler :
13
16
class_path : otx.core.schedulers.LinearWarmupSchedulerCallable
14
17
init_args :
@@ -20,13 +23,17 @@ model:
20
23
factor : 0.5
21
24
patience : 3
22
25
monitor : val/accuracy
26
+
23
27
engine :
24
28
task : MULTI_CLASS_CLS
25
29
device : auto
30
+
26
31
callback_monitor : val/accuracy
27
- data : ../../base/data/classification.yaml
32
+
33
+ data : ../../_base_/data/classification.yaml
28
34
overrides :
29
35
max_epochs : 90
36
+
30
37
callbacks :
31
38
- class_path : otx.algo.callbacks.adaptive_early_stopping.EarlyStoppingWithWarmup
32
39
init_args :
Original file line number Diff line number Diff line change
1
+ model :
2
+ class_path : otx.algo.classification.multilabel_models.timm_model.TimmModelMultilabelCls
3
+ init_args :
4
+ model_name : mobilenetv3_large
5
+ label_info : 1000
6
+
7
+ optimizer :
8
+ class_path : torch.optim.SGD
9
+ init_args :
10
+ lr : 0.0058
11
+ momentum : 0.9
12
+ weight_decay : 0.0005
13
+
14
+ scheduler :
15
+ class_path : otx.core.schedulers.LinearWarmupSchedulerCallable
16
+ init_args :
17
+ num_warmup_steps : 10
18
+ main_scheduler_callable :
19
+ class_path : lightning.pytorch.cli.ReduceLROnPlateau
20
+ init_args :
21
+ mode : max
22
+ factor : 0.5
23
+ patience : 3
24
+ monitor : val/accuracy
25
+
26
+ engine :
27
+ task : MULTI_LABEL_CLS
28
+ device : auto
29
+
30
+ callback_monitor : val/accuracy
31
+
32
+ data : ../../_base_/data/classification.yaml
33
+ overrides :
34
+ data :
35
+ task : MULTI_LABEL_CLS
36
+ data_format : datumaro
37
+
38
+ callbacks :
39
+ - class_path : otx.algo.callbacks.adaptive_early_stopping.EarlyStoppingWithWarmup
40
+ init_args :
41
+ patience : 5
You can’t perform that action at this time.
0 commit comments