diff --git a/python/paddle/fluid/reader.py b/python/paddle/fluid/reader.py index 1cb76b1f3905938dd8de55498fc8bfd83d95251f..84ccba98e60408baa5465051e971622ba62d2be6 100644 --- a/python/paddle/fluid/reader.py +++ b/python/paddle/fluid/reader.py @@ -366,9 +366,6 @@ class DataLoader(object): self.dataset_kind = _DatasetKind.MAP if batch_sampler is not None: - assert isinstance(batch_sampler, BatchSampler), \ - "batch_sampler should be None or subclass instance " \ - "of paddle.io.BatchSampler" assert batch_size == 1 and not shuffle and not drop_last, \ "batch_size/shuffle/drop_last should not be set when " \ "batch_sampler is given" diff --git a/python/paddle/fluid/tests/unittests/test_multiprocess_dataloader_exception.py b/python/paddle/fluid/tests/unittests/test_multiprocess_dataloader_exception.py index 74fe359cd7d597def5dd62e973abf001af05b81c..1bda6edfecf1c78526d25b1428f33505b3a58b2c 100644 --- a/python/paddle/fluid/tests/unittests/test_multiprocess_dataloader_exception.py +++ b/python/paddle/fluid/tests/unittests/test_multiprocess_dataloader_exception.py @@ -80,14 +80,6 @@ class TestDataLoaderAssert(unittest.TestCase): except AssertionError: pass - # batch_sampler is not instance of BatchSampler - try: - loader = DataLoader( - dataset=dataset, places=place, batch_sampler=dataset) - self.assertTrue(False) - except AssertionError: - pass - # set batch_sampler and shuffle/batch_size/drop_last try: loader = DataLoader(