mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity
784b96c059
DL-Art-School/codes/models
History
James Betker 784b96c059 Misc options to add support for training stylegan2-rosinality models: 
- Allow image_folder_dataset to normalize inbound images
- ExtensibleTrainer can denormalize images on the output path
- Support .webp - an output from LSUN
- Support logistic GAN divergence loss
- Support stylegan2 TF weight extraction for discriminator
- New injector that produces latent noise (with separated paths)
- Modify FID evaluator to be operable with rosinality-style GANs
2021-02-08 08:09:21 -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 Update flownet submodule 2020-10-24 11:59:00 -06:00
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 Misc options to add support for training stylegan2-rosinality models: 2021-02-08 08:09:21 -07:00
switched_conv Revert 2021-02-05 08:43:07 -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 Back to groupnorm 2021-02-05 08:42:11 -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 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