yjh0410 2 лет назад
Родитель
Сommit
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