rockpool.nn.modules.torch.dropout

A module implementing random dropout of neurons and time steps

Classes

TimeStepDropout(*args, **kwargs)

For each gradient update time steps are dropped with a given probability.

UnitDropout(*args, **kwargs)

For each gradient update units are dropped with a given probability.