flambe.sampler.sampler
¶
Module Contents¶
-
class
flambe.sampler.sampler.
Sampler
[source]¶ Bases:
flambe.compile.Component
Base Sampler interface.
Objects implementing this interface should implement two methods:
- sample: takes a set of data and returns an iterator
- lenght: takes a set of data and return the length of the
- iterator that would be given by the sample method
Sampler objects are used inside the Trainer to provide the data to the models. Note that pushing the data to the appropriate device is usually done inside the Trainer.
-
sample
(self, data: Sequence[Sequence[torch.Tensor]], n_epochs: int = 1)[source]¶ Sample from the list of features and yields batches.
Parameters: - data (Sequence[Sequence[torch.Tensor, ..]]) – The input data to sample from
- n_epochs (int, optional) – The number of epochs to run in the output iterator.
Yields: Iterator[Tuple[Tensor]] – A batch of data, as a tuple of Tensors