Unverified 提交 3f152e58 authored 作者: Ayush Chaurasia's avatar Ayush Chaurasia 提交者: GitHub

Update wandb_utils.py (#5908)

上级 fa05f8c9
......@@ -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):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论