.. _reference:

=============
API reference
=============

.. _api_options:

Data loaders
============

.. autoclass:: poptorch_geometric.dataloader.DataLoader

.. autoclass:: poptorch_geometric.dataloader.FixedSizeDataLoader

.. autoclass:: poptorch_geometric.pyg_dataloader.FixedSizeStrategy

.. autoclass:: poptorch_geometric.pyg_dataloader.OverSizeStrategy

Cluster data loaders
====================

.. autoclass:: poptorch_geometric.cluster_loader.FixedSizeClusterLoader

Collators
=========

.. autoclass:: poptorch_geometric.collate.FixedSizeCollater

Batch samplers
==============

.. autoclass:: poptorch_geometric.stream_packing_sampler.StreamPackingSampler

Fixed size options
==================

.. autoclass:: poptorch_geometric.fixed_size_options.FixedSizeOptions