Explorar el Código

use RTMTrainer for YOLOvx

yjh0410 hace 2 años
padre
commit
ff11208332
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  1. 1 1
      config/model_config/yolovx_config.py

+ 1 - 1
config/model_config/yolovx_config.py

@@ -119,7 +119,7 @@ yolovx_cfg = {
         # ---------------- Train config ----------------
         # training configuration
         'no_aug_epoch': 20,
-        'trainer_type': 'yolo',
+        'trainer_type': 'rtmdet',
         # optimizer
         'optimizer': 'adamw',      # optional: sgd, adam, adamw
         'momentum': None,          # SGD: 0.937;    AdamW: invalid