Hugging Face's logo

Spaces:
henry000
/
YOLO
Running

App Files Community
henry000 commited on
Commit
46ebaf7
·
1 Parent(s): 4be6676

🚚 [Rename] EMA enabled -> enable

Browse files
Files changed (2) hide show
  1. yolo/config/config.py +1 -1
  2. yolo/config/task/train.yaml +1 -1
yolo/config/config.py CHANGED
@@ -97,7 +97,7 @@ class SchedulerConfig:
97
 
98
  @dataclass
99
  class EMAConfig:
100
- enabled: bool
101
  decay: float
102
 
103
 
 
97
 
98
  @dataclass
99
  class EMAConfig:
100
+ enable: bool
101
  decay: float
102
 
103
 
yolo/config/task/train.yaml CHANGED
@@ -50,5 +50,5 @@ scheduler:
50
  end_factor: 0.01
51
 
52
  ema:
53
- enabled: true
54
  decay: 0.995
 
50
  end_factor: 0.01
51
 
52
  ema:
53
+ enable: true
54
  decay: 0.995