Flat Hilber Bayesian Inference (FHVI)

[PyTorch] Code for the paper - 'Improving Generalization with Flat Hilbert Bayesian Inference'

Includes implementation of FHBI, some configs, and log files.

Requirements

• Python 3.8+
• pip install -r requirements.txt

TODO:

• Double-check and finalize the code and config
• Update the README, upload checkpoint (if applicable)