未验证 提交 d2e50c1e 编写于 作者: C cuicheng01 提交者: GitHub

Update __init__.py

上级 fcc1b857
......@@ -41,7 +41,7 @@ def build_dataloader(config, mode, device, seed=None):
config_dataset = config[mode]['dataset']
config_dataset = copy.deepcopy(config_dataset)
dataset_name = config_dataset.pop('name')
if 'batch_transform_ops' in config_dataset:h
if 'batch_transform_ops' in config_dataset:
batch_transform = config_dataset.pop('batch_transform_ops')
else:
batch_transform = None
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册