[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 -2
View File
@@ -126,8 +126,7 @@ class WandbLogger():
if not opt.resume:
wandb_data_dict = self.check_and_upload_dataset(opt) if opt.upload_dataset else data_dict
# Info useful for resuming from artifacts
self.wandb_run.config.opt = vars(opt)
self.wandb_run.config.data_dict = wandb_data_dict
self.wandb_run.config.update({'opt': vars(opt), 'data_dict': data_dict}, allow_val_change=True)
self.data_dict = self.setup_training(opt, data_dict)
if self.job_type == 'Dataset Creation':
self.data_dict = self.check_and_upload_dataset(opt)