lightning-uq-box
lightning-uq-box copied to clipboard
Refactor Posthoc
This PR refactors Post-Hoc methods to be called with .fit() instead of .validate(). This requires more checks to make sure the model is in eval mode and that no gradients are computed, so offset the things that are automatically turned on by lightning. However, it does make the API more consistent.