diff --git a/python/paddle/v2/reader/decorator.py b/python/paddle/v2/reader/decorator.py index d83e2f4577f20bcd2220a1a29b394d925126922e..5687f118ce277928d16f4ea48907cee120c73b47 100644 --- a/python/paddle/v2/reader/decorator.py +++ b/python/paddle/v2/reader/decorator.py @@ -201,7 +201,7 @@ def batched(reader, batch_size): :return: the batched reader. """ - def __impl__(): + def batched_reader(): r = reader() batch = [] for instance in r: @@ -212,4 +212,4 @@ def batched(reader, batch_size): if batch: yield batch - return __impl__ + return batched_reader