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
..
byol	Add LARS optimizer & support for BYOL idiosyncrasies	2020-12-23 20:33:43 -07:00
fixup_resnet
flownet2@db2b7899ea
glean	glean mods	2020-12-19 08:26:07 -07:00
srflow	Fix bugs with srflow after refactor	2020-12-19 10:28:23 -07:00
stylegan
switched_conv@cb520afd4d
tecogan
transformers
__init__.py
arch_util.py	Change default ReLU slope to .2 BREAKS COMPATIBILITY	2020-12-19 08:28:03 -07:00
discriminator_vgg_arch.py
feature_arch.py
ProgressiveSrg_arch.py
ResGen_arch.py
resnet_with_checkpointing.py	Add LARS optimizer & support for BYOL idiosyncrasies	2020-12-23 20:33:43 -07:00
RRDBNet_arch.py
spinenet_arch.py
srg2_classic.py
SwitchedResidualGenerator_arch.py