From 2a417ca9a9e87abfa223325cc74821c3ca9999a5 Mon Sep 17 00:00:00 2001 From: runame Date: Thu, 31 Jul 2025 15:40:04 -0700 Subject: [PATCH] Trigger should_validate at first step --- torchtitan/components/validate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchtitan/components/validate.py b/torchtitan/components/validate.py index 7f8b848c7f..12694b25b6 100644 --- a/torchtitan/components/validate.py +++ b/torchtitan/components/validate.py @@ -28,7 +28,7 @@ def validate(self, model_parts: list[nn.Module]) -> dict[str, float]: raise NotImplementedError("validate method not implemented") def should_validate(self, step: int) -> bool: - return step % self.job_config.validation.freq == 0 + return step == 1 or step % self.job_config.validation.freq == 0 class Validator(BaseValidator):