fix(wandb): key messed in wandb option parsing (#1314)
Browse files- 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
|
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'")
|