Skip to content

Commit 284db42

Browse files
committed
feat: add wandb offline mode and custom wandb directory support
1 parent c60111a commit 284db42

1 file changed

Lines changed: 0 additions & 3 deletions

File tree

specforge/tracker.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -142,9 +142,6 @@ def __init__(self, args, output_dir: str):
142142
if args.wandb_dir is None:
143143
args.wandb_dir = self._default_wandb_dir()
144144
os.makedirs(args.wandb_dir, exist_ok=True)
145-
if args.wandb_offline:
146-
os.environ["WANDB_MODE"] = "offline"
147-
os.environ["WANDB_DIR"] = args.wandb_dir
148145

149146
if not args.wandb_offline:
150147
wandb.login(key=args.wandb_key)

0 commit comments

Comments
 (0)