Update wandb_utils.py (#5908)
This commit is contained in:
1 parent
fa05f8c977
commit
3f152e5807
1 file changed
+1
@@ -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):
|
||||
|
||||
Reference in new issue
Block a user