From 3f152e58074514b2531cb43fb57db380e085cd09 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 2d6133a..221d3c8 100644 --- a/utils/loggers/wandb/wandb_utils.py +++ b/utils/loggers/wandb/wandb_utils.py @@ -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):