norse.torch.module.regularization

norse.torch.module.regularization¶

This module contains ``torch.nn.Module``s for regularisation operations on spiking layers, where it can be desirable to regularise spikes, membrane parameters, or other properties over time.

Classes

RegularizationCell(accumulator, state)

A regularisation cell that accumulates some state (for instance number of spikes) for each forward step, which can later be applied to a loss term.