Revert "debug"

This reverts commit afc9b4c6.

Revert "debug"
This reverts commit afc9b4c6.
753270ab · Tingquan Gao · 876732fb · 753270ab
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppcls/data/__init__.py ppcls/data/__init__.py +1 -1

未找到文件。
--- a/ppcls/data/__init__.py
+++ b/ppcls/data/__init__.py
@@ -118,7 +118,7 @@ def build_dataloader(config, *mode, seed=None):
        config_dataset = copy.deepcopy(config_dataset)
        dataset_name = config_dataset.pop('name')
        if 'batch_transform_ops' in config_dataset:
-            batch_transform = config_dataset.pop('batch_transform_ops')
+            batch_transform = config_dataset['batch_transform_ops']
        else:
            batch_transform = None