James Betker 1a2b9fa130 Get rid of old byol net wrapping ... Simplifies and makes this usable with DLAS' multi-gpu trainer		2021-04-27 12:48:34 -06:00
..
byol	Get rid of old byol net wrapping	2021-04-27 12:48:34 -06:00
fixup_resnet
flownet2@db2b7899ea
glean	Glean mods	2020-12-27 12:25:06 -07:00
global_convs	Misc	2021-04-21 18:09:46 -06: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
segformer	Add testing capabilities for segformer & contrastive feature	2021-04-27 09:59:50 -06:00
srflow
styled_sr	Styled SR fixes	2021-01-05 20:14:39 -07:00
stylegan	misc adjustments for stylegan	2021-04-21 18:14:17 -06:00
switched_conv	Switched conv: add conversion function with allowlist	2021-03-13 10:44:56 -07:00
tecogan
transformers	Injector auto-registration	2020-12-29 20:58:02 -07:00
vqvae	Fix some bugs, checkin work on vqvae3	2021-03-02 20:56:19 -07:00
__init__.py
arch_util.py
discriminator_vgg_arch.py	Misc fixes	2021-01-04 10:53:53 -07:00
feature_arch.py
lightweight_gan.py	Mods to support lightweight_gan model	2021-03-02 20:51:48 -07:00
ResGen_arch.py
resnet_with_checkpointing.py	Support training imagenet classifier	2021-01-11 20:09:16 -07:00
RRDBNet_arch.py	Move switched_conv logic around a bit	2021-02-02 20:41:24 -07:00
spinenet_arch.py
weighted_conv_resnet.py	Support training imagenet classifier	2021-01-11 20:09:16 -07:00