debug

ppcls/data/__init__.py

@@ -106,14 +106,14 @@ def build_dataloader(config, *mode, seed=None):
     # build dataset
     if use_dali:
         from ppcls.data.dataloader.dali import dali_dataloader
-        return dali_dataloader(
+        data_loader = dali_dataloader(
             dataloader_config,
             mode[-1],
             paddle.device.get_device(),
             num_threads=num_workers,
             seed=seed,
             enable_fuse=True)
-
+    else:
         config_dataset = dataloader_config['dataset']
         config_dataset = copy.deepcopy(config_dataset)
         dataset_name = config_dataset.pop('name')
@@ -135,12 +135,14 @@ def build_dataloader(config, *mode, seed=None):
             shuffle = config_sampler["shuffle"]
         else:
             sampler_name = config_sampler.pop("name")
-        sampler_argspec = inspect.getargspec(eval(sampler_name).__init__).args
+            sampler_argspec = inspect.getargspec(eval(sampler_name)
+                                                 .__init__).args
             if "total_epochs" in sampler_argspec:
                 config_sampler.update({"total_epochs": epochs})
             batch_sampler = eval(sampler_name)(dataset, **config_sampler)
 
-    logger.debug("build batch_sampler({}) success...".format(batch_sampler))
+        logger.debug("build batch_sampler({}) success...".format(
+            batch_sampler))
 
         # build batch operator
         def mix_collate_fn(batch):

ppcls/metric/__init__.py

@@ -15,6 +15,7 @@
 import copy
 from collections import OrderedDict
 
+from ..utils import logger
 from .avg_metrics import AvgMetrics
 from .metrics import TopkAcc, mAP, mINP, Recallk, Precisionk
 from .metrics import DistillationTopkAcc