Jelajahi Sumber

use RTMTrainer for YOLOvx

yjh0410 2 tahun lalu
induk
melakukan
ff11208332
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  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