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