From 399b52e854958041f3d7ed07736d63dcec8fd01b Mon Sep 17 00:00:00 2001 From: Ayush Chaurasia Date: Tue, 7 Dec 2021 20:39:11 +0530 Subject: [PATCH] Update wandb_utils.py (#5908) --- utils/loggers/wandb/wandb_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/utils/loggers/wandb/wandb_utils.py b/utils/loggers/wandb/wandb_utils.py index 2d6133ab94c5..221d3c88c56e 100644 --- a/utils/loggers/wandb/wandb_utils.py +++ b/utils/loggers/wandb/wandb_utils.py @@ -186,6 +186,7 @@ def __init__(self, opt, run_id=None, job_type='Training'): 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):