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

fix(wandb): key messed in wandb option parsing
Megvii-BaseDetection · May 17, 2022 · 11ca4b0 · 11ca4b0
1 parent c8d12a5
commit 11ca4b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yolox/core/trainer.py b/yolox/core/trainer.py
@@ -183,7 +183,7 @@ def before_train(self):
                 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'")