ksuit.data.pipeline.batch_processor

Classes

Module Contents

class ksuit.data.pipeline.batch_processor.BatchProcessor
abstractmethod denormalize(key, value)

Inverts the normalization from the __call__ method of a single item in the batch. If nothing needs to be done for the denormalization, this method should simply return the passed key/value.

Parameters:
  • key (str) – The name of the item.

  • value (torch.Tensor) – The value of the item.

Returns:

The (potentially) back-mapped name and the (potentially) denormalized value.

Return type:

(key, value)