James Betker 036684893e Add LARS optimizer & support for BYOL idiosyncrasies ... - Added LARS and SGD optimizer variants that support turning off certain features for BN and bias layers - Added a variant of pytorch's resnet model that supports gradient checkpointing. - Modify the trainer infrastructure to support above - Fix bug with BYOL (should have been nonfunctional)		2020-12-23 20:33:43 -07:00
..
__init__.py
byol_model_wrapper.py	Add LARS optimizer & support for BYOL idiosyncrasies	2020-12-23 20:33:43 -07:00
byol_structural.py