fix(config): 修正 SGDRScheduleConfig 类名拼写错误
This commit is contained in:
@@ -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",
|
||||
]
|
||||
@@ -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."}
|
||||
|
||||
Reference in New Issue
Block a user