James Betker b123ed8a45 Add attention resnet ... Not ready for prime time, but is a first draft.		2020-05-29 20:02:10 -06:00
..
archs	Add attention resnet	2020-05-29 20:02:10 -06:00
__init__.py
base_model.py	Allow an alt_path for saving models and states	2020-05-16 09:10:51 -06:00
loss.py
lr_scheduler.py
networks.py	Separate feature extractors out, add resnet feature extractor	2020-05-28 20:26:30 -06:00
SR_model.py	Use amp in SR_model for inference	2020-05-07 21:45:33 -06:00
SRGAN_model.py	Support generators that don't output intermediary values.	2020-05-23 21:09:54 -06:00