credit.postblock.base

Classes

BasePostblock

Base class for all postblocks.

Module Contents

class credit.postblock.base.BasePostblock(*args: Any, **kwargs: Any)

Bases: torch.nn.Module

Base class for all postblocks.

Forward signature for all postblocks:

forward(batch: dict) -> dict

where batch is the reconstructed dict returned by Reconstruct:

{
    "prediction": {var_key: tensor, ...},
    "metadata":   {...},
}

forward(batch: dict) → dict

classmethod from_config(**kwargs)