fix(config): 修正 SGDRScheduleConfig 类名拼写错误

This commit is contained in:
2025-10-20 18:21:46 +08:00
parent e051005334
commit bc5ef72001
3 changed files with 4 additions and 4 deletions
+2 -2
View File
@@ -1,6 +1,6 @@
from khaosz.config.model_config import TransformerConfig
from khaosz.config.param_config import BaseModelIO, ModelParameter, Checkpoint, ParameterLoader
from khaosz.config.schedule_config import ScheduleConfig, CosineScheduleConfig, SgdrScheduleConfig
from khaosz.config.schedule_config import ScheduleConfig, CosineScheduleConfig, SGDRScheduleConfig
from khaosz.config.train_config import TrainConfig
@@ -14,5 +14,5 @@ __all__ = [
"ScheduleConfig",
"CosineScheduleConfig",
"SgdrScheduleConfig",
"SGDRScheduleConfig",
]
+1 -1
View File
@@ -59,7 +59,7 @@ class CosineScheduleConfig(ScheduleConfig):
@dataclass
class SgdrScheduleConfig(ScheduleConfig):
class SGDRScheduleConfig(ScheduleConfig):
cycle_length: int = field(
default=1000,
metadata={"help": "Length of the first cycle in steps."}