[forge] allow torchforges to set checkpoint base folder (#2131)

rakkit · web-flow · commit f3f2e8fd2419 · 2025-12-09T12:48:23.000-08:00
this PR 
1) allowing Torchforge to decide where to put the checkpoint and wandb,
etc, instead of the "current" folder
~~allowing Torchforge to decide to print / log the configs~~
diff --git a/torchtitan/experiments/forge/engine.py b/torchtitan/experiments/forge/engine.py
@@ -228,6 +228,7 @@ def __init__(self, job_config: ForgeJobConfig):
                 if self.train_spec.state_dict_adapter
                 else None
             ),
+            base_folder=job_config.job.dump_folder,
         )
 
         loss_parallel_enabled = (

Original file line number	Diff line number	Diff line change
`@@ -228,6 +228,7 @@ def __init__(self, job_config: ForgeJobConfig):`
`228`	`228`	`if self.train_spec.state_dict_adapter`
`229`	`229`	`else None`
`230`	`230`	`),`
	`231`	`+ base_folder=job_config.job.dump_folder,`
`231`	`232`	`)`
`232`	`233`
`233`	`234`	`loss_parallel_enabled = (`