DL-Art-School/codes/models
James Betker bba283776c Enable find_unused_parameters for DistributedDataParallel
attention_norm has some parameters which are not used to compute grad,
which is causing failures in the distributed case.
2020-07-23 09:08:13 -06:00
..
archs Add switched discriminator 2020-07-22 20:52:59 -06:00
__init__.py Implement downsample GAN 2020-04-24 00:00:46 -06:00
base_model.py Fix inverse temperature curve logic and add upsample factor 2020-06-19 09:18:30 -06:00
loss.py Add u-net discriminator with feature output 2020-07-16 10:10:09 -06:00
lr_scheduler.py Huge set of mods to support progressive generator growth 2020-07-18 14:18:48 -06:00
networks.py Add switched discriminator 2020-07-22 20:52:59 -06:00
SR_model.py Allow validating in batches, remove val size limit 2020-06-02 08:41:22 -06:00
SRGAN_model.py Enable find_unused_parameters for DistributedDataParallel 2020-07-23 09:08:13 -06:00