Select samples sequentially. More...

Inheritance diagram for torch::data::samplers::DistributedSequentialSampler:

Public Member Functions
TORCH_API	DistributedSequentialSampler (size_t size, size_t num_replicas=1, size_t rank=0, bool allow_duplicates=true)

TORCH_API void	reset (optional< size_t > new_size=nullopt) override
	Resets the `DistributedSequentialSampler` to a new set of indices.

TORCH_API optional< std::vector< size_t > >	next (size_t batch_size) override
	Returns the next batch of indices.

TORCH_API void	save (serialize::OutputArchive &archive) const override
	Serializes the `DistributedSequentialSampler` to the `archive`.

TORCH_API void	load (serialize::InputArchive &archive) override
	Deserializes the `DistributedSequentialSampler` from the `archive`.

TORCH_API size_t	index () const noexcept
	Returns the current index of the `DistributedSequentialSampler`.

Public Member Functions inherited from torch::data::samplers::DistributedSampler<>
TORCH_API	DistributedSampler (size_t size, size_t num_replicas=1, size_t rank=0, bool allow_duplicates=true)

void	set_epoch (size_t epoch)
	Set the epoch for the current enumeration. More...

size_t	epoch () const

Additional Inherited Members
Public Types inherited from torch::data::samplers::Sampler< std::vector< size_t > >
using	BatchRequestType = std::vector< size_t >

Protected Member Functions inherited from torch::data::samplers::DistributedSampler<>
size_t	local_sample_count ()

Protected Attributes inherited from torch::data::samplers::DistributedSampler<>
size_t	size_

size_t	num_replicas_

size_t	rank_

size_t	epoch_

bool	allow_duplicates_

Detailed Description

Select samples sequentially.

Definition at line 100 of file distributed.h.

The documentation for this class was generated from the following files:

Public Member Functions