ilessio-aiflowlab commited on
Commit
1f6d9b8
·
verified ·
1 Parent(s): d96812b

Upload configs/msrs_b2.toml with huggingface_hub

Browse files
Files changed (1) hide show
  1. configs/msrs_b2.toml +57 -0
configs/msrs_b2.toml ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # RTFDNet Training Config — FMB dataset, MiT-B2 backbone
2
+ # Paper: RTFDNet (ArXiv 2603.09149)
3
+
4
+ [model]
5
+ variant = "mit_b2"
6
+ channels = 256
7
+ pretrained = "models/mit_b2.pth"
8
+ drop_path_rate = 0.1
9
+ dropout_ratio = 0.1
10
+ gradient_checkpointing = false
11
+
12
+ [data]
13
+ dataset = "msrs"
14
+ data_root = "data/ade_msrs"
15
+ num_classes = 9
16
+ crop_size = [480, 640]
17
+ num_workers = 4
18
+ pin_memory = true
19
+
20
+ [training]
21
+ batch_size = "auto"
22
+ learning_rate = 3e-5
23
+ weight_decay = 0.01
24
+ optimizer = "adamw"
25
+ betas = [0.9, 0.999]
26
+ max_epochs = 300
27
+ precision = "fp16"
28
+ gradient_accumulation = 1
29
+ max_grad_norm = 1.0
30
+ seed = 42
31
+ head_lr_mult = 10.0
32
+
33
+ [scheduler]
34
+ type = "warmup_cosine"
35
+ warmup_ratio = 0.05
36
+ min_lr = 1e-5
37
+ T_max = 5
38
+
39
+ [losses]
40
+ ce_weight = 0.5
41
+ modal_ce_weight = 1.0
42
+ akd_weight = 0.5
43
+ region_l1_weight = 1.0
44
+
45
+ [checkpoint]
46
+ save_every_n_epochs = 10
47
+ keep_top_k = 2
48
+ metric = "mIoU"
49
+ mode = "max"
50
+
51
+ [early_stopping]
52
+ enabled = true
53
+ patience = 30
54
+ min_delta = 0.001
55
+
56
+ [validation]
57
+ interval = 5