Преглед на файлове

use RTMTrainer for YOLOvx

yjh0410 преди 2 години
родител
ревизия
ff11208332
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  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