DataLoader

    注解

    框架保证DataLoader的数据加载顺序与用户提供的数据源读取顺序一致。

    创建一个DataLoader对象用于加载Python生成器产生的数据。数据会由Python线程预先读取，并异步送入一个队列中。

    本方法创建的DataLoader对象提供了3个方法设置数据源，分别是 set_sample_generator , set_sample_list_generator 和 。请查阅下述示例代码了解它们的使用方法。

    如果iterable = True，本方法创建的DataLoader对象时一个Python生成器，可以for-range的方法循环迭代。

    如果iterable = False，本方法创建的DataLoader对象提供 start() 和 reset() 方法控制数据读取过程。此模式用于兼容 的使用方式。用户可使用iterable = False模式，方便地将 fluid.layers.py_reader 的代码迁移至 fluid.io.DataLoader 。

    返回

    被创建的DataLoader对象

    返回类型

    loader (DataLoader)

    代码示例 1

    代码示例 2

    from_dataset(dataset, places, drop_last=True)

    参数

    • dataset (InMemoryDataset|QueueDataset) - Dataset对象。
    • drop_last (bool) - 是否丢弃最后样本数量不足batch size的batch。若drop_last = True则丢弃，若drop_last = False则不丢弃。

    返回

    被创建的DataLoader对象，可以for-range的方式循环迭代

    返回类型

    loader (DataLoader)

    代码示例