DL-Art-School/codes/models
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 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-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 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 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 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 Fix some bugs, checkin work on vqvae3 2021-03-02 20:56:19 -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
lightweight_gan.py Mods to support lightweight_gan model 2021-03-02 20:51:48 -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 Move switched_conv logic around a bit 2021-02-02 20:41:24 -07:00
spinenet_arch.py Migrate generators to dynamic model registration 2020-12-24 23:02:10 -07:00
weighted_conv_resnet.py Support training imagenet classifier 2021-01-11 20:09:16 -07:00