From 5f578f92a72c5c735608dbf28bad7d9fb7ac461b Mon Sep 17 00:00:00 2001 From: Frankstein <20307140057@fudan.edu.cn> Date: Sat, 24 Aug 2024 16:06:17 +0800 Subject: [PATCH] fix merge error --- src/lm_saes/runner.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/lm_saes/runner.py b/src/lm_saes/runner.py index 5f379bf..6ffc2df 100644 --- a/src/lm_saes/runner.py +++ b/src/lm_saes/runner.py @@ -98,8 +98,8 @@ def language_model_sae_runner(cfg: LanguageModelSAETrainingConfig): sae.train_finetune_for_suppression_parameters() if is_master(): - cfg.sae.save_hyperparameters(os.path.join(cfg.exp_result_dir, cfg.exp_name)) - cfg.lm.save_lm_config(os.path.join(cfg.exp_result_dir, cfg.exp_name)) + cfg.sae.save_hyperparameters(cfg.exp_result_path) + cfg.lm.save_lm_config(cfg.exp_result_path) if cfg.wandb.log_to_wandb and is_master(): wandb_config: dict = {