James Betker 574e7e882b Fix up OOM issues when running a disjoint D update ratio and megabatches		2020-05-06 17:25:25 -06:00
..
archs	Fix up OOM issues when running a disjoint D update ratio and megabatches	2020-05-06 17:25:25 -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
lr_scheduler.py
networks.py	Implement ResGen arch	2020-05-05 11:59:46 -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