Ayush Chaurasia
commited on
Update wandb_utils.py (#5908)
Browse files
utils/loggers/wandb/wandb_utils.py
CHANGED
@@ -186,6 +186,7 @@ class WandbLogger():
|
|
186 |
self.setup_training(opt)
|
187 |
|
188 |
if self.job_type == 'Dataset Creation':
|
|
|
189 |
self.data_dict = self.check_and_upload_dataset(opt)
|
190 |
|
191 |
def check_and_upload_dataset(self, opt):
|
|
|
186 |
self.setup_training(opt)
|
187 |
|
188 |
if self.job_type == 'Dataset Creation':
|
189 |
+
self.wandb_run.config.update({"upload_dataset": True})
|
190 |
self.data_dict = self.check_and_upload_dataset(opt)
|
191 |
|
192 |
def check_and_upload_dataset(self, opt):
|