James Betker c74b9ee2e4 Add a way to disable grad on portions of the generator graph to save memory		2020-07-22 11:40:42 -06:00
..
archs	Add a way to disable grad on portions of the generator graph to save memory	2020-07-22 11:40:42 -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	Allow feature discriminator unet to only output closest layer to feature output	2020-07-19 19:05:08 -06:00
SR_model.py	Allow validating in batches, remove val size limit	2020-06-02 08:41:22 -06:00
SRGAN_model.py	Add convert_model.py and a hacky way to add extra layers to a model	2020-07-22 11:39:45 -06:00