James Betker f894ba8f98 Add SPSR_module ... This is a port from the SPSR repo, it's going to need a lot of work to be properly integrated but as of this commit it at least runs.		2020-08-01 22:02:54 -06:00
..
archs	Update how attention_maps are created	2020-08-01 20:23:46 -06:00
SPSR_modules	Add SPSR_module	2020-08-01 22:02:54 -06:00
__init__.py	Add SPSR_module	2020-08-01 22:02:54 -06:00
base_model.py	Add feature_model for training custom feature nets	2020-07-31 11:20:39 -06:00
feature_model.py	Add feature_model for training custom feature nets	2020-07-31 11:20:39 -06:00
loss.py	Add FDPL Loss	2020-07-30 20:47:57 -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	Enable disjoint feature networks	2020-07-31 16:29:47 -06:00
SPSR_model.py	Add SPSR_module	2020-08-01 22:02:54 -06:00
SPSR_networks.py	Add SPSR_module	2020-08-01 22:02:54 -06:00
SR_model.py	Allow validating in batches, remove val size limit	2020-06-02 08:41:22 -06:00
SRGAN_model.py	More torch 1.6 fixes	2020-07-31 17:03:20 -06:00