stephanxu commited on
Commit
e6b9c0d
·
1 Parent(s): 32fad93

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

Browse files
Files changed (1) hide show
  1. yolox/core/trainer.py +1 -1
yolox/core/trainer.py CHANGED
@@ -183,7 +183,7 @@ class Trainer:
183
  wandb_params = dict()
184
  for k, v in zip(self.args.opts[0::2], self.args.opts[1::2]):
185
  if k.startswith("wandb-"):
186
- wandb_params.update({k.lstrip("wandb-"): v})
187
  self.wandb_logger = WandbLogger(config=vars(self.exp), **wandb_params)
188
  else:
189
  raise ValueError("logger must be either 'tensorboard' or 'wandb'")
 
183
  wandb_params = dict()
184
  for k, v in zip(self.args.opts[0::2], self.args.opts[1::2]):
185
  if k.startswith("wandb-"):
186
+ wandb_params.update({k[len("wandb-"):]: v})
187
  self.wandb_logger = WandbLogger(config=vars(self.exp), **wandb_params)
188
  else:
189
  raise ValueError("logger must be either 'tensorboard' or 'wandb'")