probly.representer.sampler.torch

Sampling preparation for torch.

Functions

register_forced_train_mode

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