summaryrefslogtreecommitdiff
path: root/ep_run/casc_eq_train.py
diff options
context:
space:
mode:
authorYuren Hao <yurenh2@illinois.edu>2026-07-14 12:53:22 -0500
committerYuren Hao <yurenh2@illinois.edu>2026-07-14 12:53:22 -0500
commit8a9db1232a1bca01900f4113bb68666229384512 (patch)
tree40ecb6702b1151a71b67f7eadaa549e42a6c51b8 /ep_run/casc_eq_train.py
parentbba77d7a45ac29f5a640dac26c097752db1a59bb (diff)
wandb -> team entity eqprop-llm-training (trainers + full-history replay script)
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_014FAPDWQ49M5Ye3NpTndTpn
Diffstat (limited to 'ep_run/casc_eq_train.py')
-rw-r--r--ep_run/casc_eq_train.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/ep_run/casc_eq_train.py b/ep_run/casc_eq_train.py
index faf827e..2e221f6 100644
--- a/ep_run/casc_eq_train.py
+++ b/ep_run/casc_eq_train.py
@@ -529,7 +529,7 @@ wb = None
if args.wandb and RANK == 0:
try:
import wandb as _w
- wb = _w.init(project=args.wandb, name=args.wandb_run or args.tag, id=args.wandb_run or args.tag,
+ wb = _w.init(entity='eqprop-llm-training', project=args.wandb, name=args.wandb_run or args.tag, id=args.wandb_run or args.tag,
resume='allow', config=vars(args))
except Exception as e:
print(f'[wandb] disabled ({e})', flush=True)