update model saving directory

taufeeque9 · web-flow · commit 9bcbcf49cf66 · 2024-02-12T13:07:23.000+05:30
diff --git a/codebook_features/train_codebook.py b/codebook_features/train_codebook.py
@@ -134,7 +134,8 @@ def main(cfg):
             dir=training_args.output_dir,
         )
         wandb_initilized = True
-        training_args.output_dir = wandb.run.dir + "/train_output"
+        training_args.output_dir = pathlib.Path(wandb.run.dir).parent / "train_output"
+        training_args.output_dir = str(training_args.output_dir)
 
     model = transformers.AutoModelForCausalLM.from_pretrained(
         model_args.model_name_or_path,

Original file line number	Diff line number	Diff line change
`@@ -134,7 +134,8 @@ def main(cfg):`
`134`	`134`	`dir=training_args.output_dir,`
`135`	`135`	`)`
`136`	`136`	`wandb_initilized = True`
`137`		`- training_args.output_dir = wandb.run.dir + "/train_output"`
	`137`	`+ training_args.output_dir = pathlib.Path(wandb.run.dir).parent / "train_output"`
	`138`	`+ training_args.output_dir = str(training_args.output_dir)`
`138`	`139`
`139`	`140`	`model = transformers.AutoModelForCausalLM.from_pretrained(`
`140`	`141`	`model_args.model_name_or_path,`