提交 afc9b4c6 编写于 作者: G gaotingquan 提交者: Wei Shengyu

debug

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