Source code for rindti.layers.base_layer

from pytorch_lightning import LightningModule


[docs]class BaseLayer(LightningModule):
    """Base class for all layers."""

    def __init__(self, *args, **kwargs):
        super().__init__()

[docs]    def forward(self, *args, **kwargs):
        """"""
        raise NotImplementedError()