jiarui_li/DeepHealth
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore(train): include model_name in saved config filename (config_{model}_n_embd_...json)

Browse Source
This commit is contained in:
Jiarui Li
2025-10-22 17:56:23 +08:00
parent 4d1fc63667
commit 02be03f784
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
train.py
View File

@@ -85,7 +85,8 @@ def main():
checkpoint_filename = f"best_model_checkpoint_{model_suffix}.pt" checkpoint_filename = f"best_model_checkpoint_{model_suffix}.pt"
# --- 0. Save Configuration --- # --- 0. Save Configuration ---
config_filename = f"config_n_embd_{config.n_embd}_n_layer_{config.n_layer}_n_head_{config.n_head}.json" # Include model class in config filename for clarity/distinction across architectures
config_filename = f"config_{config.model_name}_n_embd_{config.n_embd}_n_layer_{config.n_layer}_n_head_{config.n_head}.json"
config_dict = {k: v for k, v in vars(config).items() if not k.startswith('__')} config_dict = {k: v for k, v in vars(config).items() if not k.startswith('__')}
with open(config_filename, 'w') as f: with open(config_filename, 'w') as f:
json.dump(config_dict, f, indent=4) json.dump(config_dict, f, indent=4)