|
|
@@ -22,7 +22,7 @@ class YolofBaseConfig(object):
|
|
|
self.neck_expand_ratio = 0.25
|
|
|
self.neck_dilations = [2, 4, 6, 8]
|
|
|
## Head
|
|
|
- self.head_dim = 512
|
|
|
+ self.head_dim = 512
|
|
|
self.num_cls_head = 2
|
|
|
self.num_reg_head = 4
|
|
|
|
|
|
@@ -48,27 +48,27 @@ class YolofBaseConfig(object):
|
|
|
self.focal_loss_gamma = 2.0
|
|
|
self.loss_cls = 1.0
|
|
|
self.loss_reg = 1.0
|
|
|
- self.loss_ctn = 1.0
|
|
|
|
|
|
# ---------------- ModelEMA config ----------------
|
|
|
- self.use_ema = True
|
|
|
+ self.use_ema = False
|
|
|
self.ema_decay = 0.9998
|
|
|
self.ema_tau = 2000
|
|
|
|
|
|
# ---------------- Optimizer config ----------------
|
|
|
- self.trainer = 'yolo'
|
|
|
+ self.trainer = 'simple'
|
|
|
self.optimizer = 'adamw'
|
|
|
- self.base_lr = 0.001 # base_lr = per_image_lr * batch_size
|
|
|
- self.min_lr_ratio = 0.01 # min_lr = base_lr * min_lr_ratio
|
|
|
+ self.base_lr = 0.0001 # base_lr = per_image_lr * batch_size
|
|
|
+ self.min_lr_ratio = 0.01 # min_lr = base_lr * min_lr_ratio
|
|
|
+ self.bk_lr_ratio = 1.0
|
|
|
self.batch_size_base = 64
|
|
|
self.momentum = 0.9
|
|
|
- self.weight_decay = 0.05
|
|
|
- self.clip_max_norm = 35.0
|
|
|
- self.warmup_bias_lr = 0.1
|
|
|
- self.warmup_momentum = 0.8
|
|
|
+ self.weight_decay = 0.0001
|
|
|
+ self.clip_max_norm = 10.0
|
|
|
+ self.warmup_bias_lr = 0.0
|
|
|
+ self.warmup_momentum = 0.9
|
|
|
|
|
|
# ---------------- Lr Scheduler config ----------------
|
|
|
- self.warmup_epoch = 3
|
|
|
+ self.warmup_iters = 500
|
|
|
self.lr_scheduler = "cosine"
|
|
|
self.max_epoch = 150
|
|
|
self.eval_epoch = 10
|
yjh0410