DL-Art-School/codes/models
James Betker aa0305def9 Resnet discriminator overhaul
It's been a tough day figuring out WTH is going on with my discriminators.
It appears the raw FixUp discriminator can get into an "defective" state where
they stop trying to learn and just predict as close to "0" D_fake and D_real as
possible. In this state they provide no feedback to the generator and never
recover. Adding batch norm back in seems to fix this so it must be some sort
of parameterization error.. Should look into fixing this in the future.
2020-05-06 17:27:30 -06:00
..
archs Resnet discriminator overhaul 2020-05-06 17:27:30 -06:00
__init__.py Implement downsample GAN 2020-04-24 00:00:46 -06:00
base_model.py Enable AMP optimizations & write sample train images to folder. 2020-04-21 16:28:06 -06:00
loss.py mmsr 2019-08-23 21:42:47 +08:00
lr_scheduler.py mmsr 2019-08-23 21:42:47 +08:00
networks.py Resnet discriminator overhaul 2020-05-06 17:27:30 -06:00
SR_model.py Support inference across batches, support inference on cpu, checkpoint 2020-05-04 08:48:25 -06:00
SRGAN_model.py Fix up OOM issues when running a disjoint D update ratio and megabatches 2020-05-06 17:25:25 -06:00
Video_base_model.py mmsr 2019-08-23 21:42:47 +08:00