[x]W&B: Don't resume transfer learning runs (#3604)

* Allow config cahnge

* Allow val change in wandb config

* Don't resume transfer learning runs

* Add entity in log dataset
This commit is contained in:
Ayush Chaurasia authored and GitHub committed 2021-06-21 14:00:25 +02:00
1 parent e8810a53e8
commit 75c0ff43af
3 files changed
+4 -2

No files matched your search

+1
View File
@@ -89,6 +89,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
# W&B
opt.hyp = hyp # add hyperparameters
run_id = torch.load(weights).get('wandb_id') if weights.endswith('.pt') and os.path.isfile(weights) else None
run_id = run_id if opt.resume else None # start fresh run if transfer learning
wandb_logger = WandbLogger(opt, save_dir.stem, run_id, data_dict)
loggers['wandb'] = wandb_logger.wandb
if loggers['wandb']: