James Betker bc20b4739e vqvae3 ... Changes VQVAE as so: - Reverts back to smaller codebook - Adds an additional conv layer at the highest resolution for both the encoder & decoder - Uses LeakyReLU on trunk		2021-01-29 15:24:26 -07:00
..
byol	Migrate generators to dynamic model registration	2020-12-24 23:02:10 -07:00
fixup_resnet	More refactoring	2020-12-18 09:18:34 -07:00
flownet2@db2b7899ea
glean	Glean mods	2020-12-27 12:25:06 -07:00
optical_flow	Add PWCNet for human optical flow	2021-01-25 08:25:44 -07:00
pixel_level_contrastive_learning	resnet_unet_3	2021-01-15 14:51:03 -07:00
srflow	Migrate generators to dynamic model registration	2020-12-24 23:02:10 -07:00
styled_sr	Styled SR fixes	2021-01-05 20:14:39 -07:00
stylegan	More mods to support styledsr	2021-01-04 11:32:28 -07:00
tecogan	Migrate generators to dynamic model registration	2020-12-24 23:02:10 -07:00
transformers	Injector auto-registration	2020-12-29 20:58:02 -07:00
vqvae	vqvae3	2021-01-29 15:24:26 -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	Misc fixes	2021-01-04 10:53:53 -07:00
feature_arch.py	More refactoring	2020-12-18 09:18:34 -07:00
ResGen_arch.py	More refactoring	2020-12-18 09:18:34 -07:00
resnet_with_checkpointing.py	Support training imagenet classifier	2021-01-11 20:09:16 -07:00
RRDBNet_arch.py	New switched_conv	2021-01-23 13:46:30 -07:00
spinenet_arch.py	Migrate generators to dynamic model registration	2020-12-24 23:02:10 -07:00
switched_conv_hard_routing.py	Add switch norm, up dropout rate, detach selector	2021-01-26 09:31:53 -07:00
switched_conv.py	Enable lambda visualization	2021-01-23 15:53:27 -07:00
weighted_conv_resnet.py	Support training imagenet classifier	2021-01-11 20:09:16 -07:00