probly.representation.sampling.torch_sampler

Sampling preparation for torch.

Functions

register_forced_train_mode(cls)

Register a class to be forced into train mode during sampling.

probly.representation.sampling.torch_sampler.register_forced_train_mode(cls)

Register a class to be forced into train mode during sampling.

This enables Monte Carlo sampling techniques like MC Dropout [GG16b] or DropConnect :cite: mobinyDropConnectEffective2019.

Parameters:

cls (LazyType)

Return type:

None