Unverified Commit f7401f34 authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #2648 from kshen3778/wandblogger_fix2

Init parameter modifications for WandbLogger
parents 50616ca5 6cd31861
......@@ -25,8 +25,7 @@ class WandbLogger(object):
def __init__(self,
name: Optional[str] = None,
entity: Optional[str] = None,
project: Optional[str] = None,
save_dir: Optional[str] = None,
project: Optional[str] = "deepchem",
mode: Optional[str] = "online",
id: Optional[str] = None,
resume: Optional[Union[bool, str]] = None,
......@@ -43,8 +42,6 @@ class WandbLogger(object):
an entity is a username or team name where you're sending the W&B run
project: str
the name of the project where you're sending the new W&B run
save_dir: str
path where data is saved (wandb dir by default)
mode: str
W&B online or offline mode
id: str
......@@ -67,7 +64,6 @@ class WandbLogger(object):
'Note: Model checkpoints will not be uploaded to W&B in offline mode.\n'
'Please set `mode="online"` if you need to log your model.')
self.save_dir = save_dir
self.save_run_history = save_run_history
# set wandb init arguments
......@@ -77,7 +73,6 @@ class WandbLogger(object):
entity=entity,
mode=mode,
id=id,
dir=save_dir,
resume=resume,
anonymous=anonymous)
self.wandb_init_params.update(**kwargs)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment