From fd85a4af10da8e8cd62e3475720ad5c718cd8ff9 Mon Sep 17 00:00:00 2001 From: Kaipeng Deng Date: Mon, 26 Apr 2021 10:53:43 +0800 Subject: [PATCH] fix IterableDataset doc. test=document_fix (#32544) --- python/paddle/fluid/dataloader/dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/dataloader/dataset.py b/python/paddle/fluid/dataloader/dataset.py index e46083295d1..bf3d0a81f99 100755 --- a/python/paddle/fluid/dataloader/dataset.py +++ b/python/paddle/fluid/dataloader/dataset.py @@ -97,10 +97,10 @@ class IterableDataset(Dataset): .. code-block:: python import numpy as np - from paddle.io import Dataset + from paddle.io import IterableDataset # define a random dataset - class RandomDataset(Dataset): + class RandomDataset(IterableDataset): def __init__(self, num_samples): self.num_samples = num_samples -- GitLab