fix(trainer): 修复参数传递问题和检查点保存问题

This commit is contained in:
2025-12-08 13:28:11 +08:00
parent c98b175cd5
commit c934210066
3 changed files with 28 additions and 4 deletions
+10 -1
View File
@@ -93,4 +93,13 @@ class TrainConfig:
kwargs: dict = field(
default_factory=dict,
metadata={"help": "Other arguments."}
)
)
def __post_init__(self):
self.validate()
def validate(self):
required_fields = ["model", "strategy", "dataset", "optimizer", "scheduler"]
for field_name in required_fields:
if getattr(self, field_name) is None:
raise ValueError(f"{field_name} is required.")