credit.preblock.base

Classes

BasePreblock

Base class for all preblocks. Enforces the forward signature and

Module Contents

class credit.preblock.base.BasePreblock(*args: Any, **kwargs: Any)

Bases: torch.nn.Module

Base class for all preblocks. Enforces the forward signature and provides the from_config classmethod used by the registry.

VALID_DATA_TYPES = ('input', 'target')

forward(batch: dict) → dict

classmethod from_config(**kwargs)