Ayush Chaurasia
commited on
Commit
•
3f152e5
1
Parent(s):
fa05f8c
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):
|