James Betker 0dca36946f Hard Routing mods ... - Turns out my custom convolution was RIDDLED with backwards bugs, which is why the existing implementation wasn't working so well. - Implements the switch logic from both Mixture of Experts and Switch Transformers for testing purposes.		2021-02-02 20:35:58 -07:00
..
byol
fixup_resnet
flownet2@db2b7899ea
glean
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
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
transformers
vqvae	Hard Routing mods	2021-02-02 20:35:58 -07:00
__init__.py
arch_util.py
discriminator_vgg_arch.py
feature_arch.py
ResGen_arch.py
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
switched_conv_hard_routing.py	Hard Routing mods	2021-02-02 20:35:58 -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