DL-Art-School/codes/models
James Betker bdbab65082 Allow optimizers to train separate param groups, add higher dimensional VGG discriminator
Did this to support training 512x512px networks off of a pretrained 256x256 network.
2021-01-02 15:10:06 -07:00
..
byol Migrate generators to dynamic model registration 2020-12-24 23:02:10 -07:00
fixup_resnet
flownet2@db2b7899ea
glean Glean mods 2020-12-27 12:25:06 -07:00
improve_rrdb styled_sr: fix bug when using initial_stride 2021-01-01 12:13:21 -07:00
srflow Migrate generators to dynamic model registration 2020-12-24 23:02:10 -07:00
stylegan Move discriminators to the create_model paradigm 2021-01-01 15:56:09 -07:00
switched_conv@cb520afd4d
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
__init__.py
arch_util.py
discriminator_vgg_arch.py Allow optimizers to train separate param groups, add higher dimensional VGG discriminator 2021-01-02 15:10:06 -07:00
feature_arch.py
ResGen_arch.py
resnet_with_checkpointing.py Mods to support image classification & filtering 2020-12-26 13:49:27 -07:00
RRDBNet_arch.py Migrate generators to dynamic model registration 2020-12-24 23:02:10 -07:00
spinenet_arch.py Migrate generators to dynamic model registration 2020-12-24 23:02:10 -07:00
srg2_classic.py Migrate generators to dynamic model registration 2020-12-24 23:02:10 -07:00
SwitchedResidualGenerator_arch.py Migrate generators to dynamic model registration 2020-12-24 23:02:10 -07:00