未验证 提交 11ca4b0f 编写于 作者: S Stephan 提交者: GitHub

fix(wandb): key messed in wandb option parsing (#1314)

fix(wandb): key messed in wandb option parsing
上级 c8d12a56
......@@ -183,7 +183,7 @@ class Trainer:
wandb_params = dict()
for k, v in zip(self.args.opts[0::2], self.args.opts[1::2]):
if k.startswith("wandb-"):
wandb_params.update({k.lstrip("wandb-"): v})
wandb_params.update({k[len("wandb-"):]: v})
self.wandb_logger = WandbLogger(config=vars(self.exp), **wandb_params)
else:
raise ValueError("logger must be either 'tensorboard' or 'wandb'")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册