🔧 [Update] config, add epoch, exp name

yolo/config/config.py

@@ -78,6 +78,7 @@ class LossConfig:
 
 @dataclass
 class TrainConfig:
+    epoch: int
     optimizer: OptimizerConfig
     scheduler: SchedulerConfig
     ema: EMAConfig
@@ -126,3 +127,4 @@ class Config:
     model: Model
     download: Download
     hyper: HyperConfig
+    name: str

yolo/config/config.yaml

@@ -8,4 +8,6 @@ defaults:
  - augmentation: ../data/augmentation
  - model: v9-c
  - hyper: default
- - _self_
+ - _self_
+ 
+name: v9-dev

yolo/config/hyper/default.yaml

@@ -6,6 +6,7 @@ data:
   class_num: 80
   image_size: [640, 640]
 train:
+  epoch: 10
   optimizer:
     type: Adam
     args:

yolo/tools/trainer.py

@@ -22,7 +22,7 @@ class Trainer:
         self.scheduler = get_scheduler(self.optimizer, train_cfg.scheduler)
         self.loss_fn = get_loss_function(cfg)
 
-        if train_cfg.ema.get("enabled", False):
+        if getattr(train_cfg.ema, "enabled", False):
             self.ema = EMA(model, decay=train_cfg.ema.decay)
         else:
             self.ema = None