probly.representation.sampling.flax_sampler¶
Sampling preparation for flax.
Functions
Register a class to be forced into train mode during sampling. |
Back to top