Update wandb_utils.py (#5908)

This commit is contained in:
Ayush Chaurasia authored and GitHub committed 2021-12-07 16:09:11 +01:00
1 parent fa05f8c977
commit 3f152e5807
1 file changed
+1
+1
View File
@@ -186,6 +186,7 @@ class WandbLogger():
self.setup_training(opt)
if self.job_type == 'Dataset Creation':
self.wandb_run.config.update({"upload_dataset": True})
self.data_dict = self.check_and_upload_dataset(opt)
def check_and_upload_dataset(self, opt):