Quellcode durchsuchen

use RTMTrainer for YOLOvx

yjh0410 vor 2 Jahren
Ursprung
Commit
ff11208332
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  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