Moved compute_pi from utils.py to fisher_blocks.py and added global constant...
Moved compute_pi from utils.py to fisher_blocks.py and added global constant that controls the type of pi-based damping adjustment to use. Current options are "tracenorm", which is what we were already doing and is the default, and "off". PiperOrigin-RevId: 177610677
Loading
Please sign in to comment