fastreid.data.samplers.data_sampler.InferenceSampler Class Reference

Inheritance diagram for fastreid.data.samplers.data_sampler.InferenceSampler:

Public Member Functions
	__init__ (self, int size)
	__iter__ (self)
	__len__ (self)

Protected Attributes
	_size
	_rank
	_world_size
	_local_indices

Detailed Description

Produce indices for inference.
Inference needs to run on the __exact__ set of samples,
therefore when the total number of samples is not divisible by the number of workers,
this sampler produces different number of samples on different workers.

Definition at line 58 of file data_sampler.py.

Constructor & Destructor Documentation

__init__()

fastreid.data.samplers.data_sampler.InferenceSampler.__init__	(		self,
		int	size )

Args:
    size (int): the total number of data of the underlying dataset to sample from

Definition at line 66 of file data_sampler.py.

    def __init__(self, size: int):
        """
        Args:
            size (int): the total number of data of the underlying dataset to sample from
        """
        self._size = size
        assert size > 0
        self._rank = comm.get_rank()
        self._world_size = comm.get_world_size()
 
        shard_size = (self._size - 1) // self._world_size + 1
        begin = shard_size * self._rank
        end = min(shard_size * (self._rank + 1), self._size)
        self._local_indices = range(begin, end)
 

Member Function Documentation

__iter__()

fastreid.data.samplers.data_sampler.InferenceSampler.__iter__

(

self

)

Definition at line 81 of file data_sampler.py.

    def __iter__(self):
        yield from self._local_indices
 

__len__()

fastreid.data.samplers.data_sampler.InferenceSampler.__len__

(

self

)

Definition at line 84 of file data_sampler.py.

    def __len__(self):
        return len(self._local_indices)

Member Data Documentation

_local_indices

fastreid.data.samplers.data_sampler.InferenceSampler._local_indices

protected

Definition at line 79 of file data_sampler.py.

_rank

fastreid.data.samplers.data_sampler.InferenceSampler._rank

protected

Definition at line 73 of file data_sampler.py.

_size

fastreid.data.samplers.data_sampler.InferenceSampler._size

protected

Definition at line 71 of file data_sampler.py.

_world_size

fastreid.data.samplers.data_sampler.InferenceSampler._world_size

protected

Definition at line 74 of file data_sampler.py.

---

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

• smreid/fastreid/data/samplers/data_sampler.py