tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2025-12-06 12:22:18 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

get report right

Browse Source
This commit is contained in:
Shizhe Diao 2025-10-20 11:51:01 -07:00
parent e872e798c4
commit 370de99bbf
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/base_loss.py
View File

@ -71,7 +71,7 @@ if ddp_rank == 0:
# Log to report # Log to report
from nanochat.report import get_report from nanochat.report import get_report
get_report().log(section="Base model loss", data=[ get_report(exp_name=model_tag).log(section="Base model loss", data=[
{ {
"train bpb": bpb_results["train"], "train bpb": bpb_results["train"],
"val bpb": bpb_results["val"], "val bpb": bpb_results["val"],

2
scripts/base_train.py
View File

@ -327,7 +327,7 @@ print0(f"Minimum validation bpb: {min_val_bpb:.4f}")
# Log to report # Log to report
from nanochat.report import get_report from nanochat.report import get_report
get_report().log(section="Base model training", data=[ get_report(exp_name=run).log(section="Base model training", data=[
user_config, # CLI args user_config, # CLI args
{ # stats about the training setup { # stats about the training setup
"Number of parameters": num_params, "Number of parameters": num_params,