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	More refactoring	2020-12-18 09:18:34 -07:00
flownet2@db2b7899ea	Update flownet submodule	2020-10-24 11:59:00 -06:00
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	GLEAN!	2020-12-18 16:04:19 -07:00
switched_conv@cb520afd4d	More refactoring	2020-12-18 09:18:34 -07:00
tecogan	More refactor changes	2020-12-18 09:24:31 -07:00
transformers	More refactoring	2020-12-18 09:18:34 -07:00
__init__.py	Lots of new discriminator nets	2020-11-10 16:06:54 -07:00
arch_util.py	Change default ReLU slope to .2 BREAKS COMPATIBILITY	2020-12-19 08:28:03 -07:00
discriminator_vgg_arch.py	More refactor changes	2020-12-18 09:24:31 -07:00
feature_arch.py	More refactoring	2020-12-18 09:18:34 -07:00
ProgressiveSrg_arch.py	More refactor changes	2020-12-18 09:24:31 -07:00
ResGen_arch.py	More refactoring	2020-12-18 09:18:34 -07:00
resnet_with_checkpointing.py	Add LARS optimizer & support for BYOL idiosyncrasies	2020-12-23 20:33:43 -07:00
RRDBNet_arch.py	More refactor changes	2020-12-18 09:24:31 -07:00
spinenet_arch.py	More refactor changes	2020-12-18 09:24:31 -07:00
srg2_classic.py	More refactor changes	2020-12-18 09:24:31 -07:00
SwitchedResidualGenerator_arch.py	More refactor changes	2020-12-18 09:24:31 -07:00