James Betker 26a6a5d512 Compute grad GAN loss against both the branch and final target, simplify pixel loss ... Also fixes a memory leak issue where we weren't detaching our loss stats when logging them. This stabilizes memory usage substantially.		2020-08-05 12:08:15 -06:00
..
archs	More RAGAN fixes	2020-08-05 11:03:06 -06:00
__init__.py	Integrate SPSR into SRGAN_model	2020-08-02 12:55:08 -06:00
base_model.py	Fix multistep optimizer (feeding from wrong config params)	2020-08-04 16:42:58 -06:00
feature_model.py	Add feature_model for training custom feature nets	2020-07-31 11:20:39 -06:00
loss.py	Substantial SPSR mods & fixes	2020-08-02 10:45:24 -06:00
lr_scheduler.py	Allow multi_step_lr_scheduler to load a new LR schedule when restoring state	2020-07-31 11:21:11 -06:00
networks.py	Several new spsr nets	2020-08-05 10:01:24 -06:00
SPSR_model.py	Integrate SPSR into SRGAN_model	2020-08-02 12:55: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	Compute grad GAN loss against both the branch and final target, simplify pixel loss	2020-08-05 12:08:15 -06:00