.. _dataset_prefetching:

.. module:: mlip.data.helpers.data_prefetching

Data parallelization and batch prefetching
==========================================

.. autoclass:: ParallelGraphDataset

    .. automethod:: __init__

    .. automethod:: __iter__

    .. automethod:: __len__

.. autoclass:: PrefetchIterator

    .. automethod:: __init__

    .. automethod:: __iter__

    .. automethod:: __len__