Merge pull request #2648 from kshen3778/wandblogger_fix2 (f7401f34) · Commits · 钟慕尧 / deepchem

deepchem/models/wandblogger.py

+1 −6

Original line number	Diff line number	Diff line
		@@ -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)