James Betker c8da78966b Substantial SPSR mods & fixes ... - Added in gradient accumulation via mega-batch-factor - Added AMP - Added missing train hooks - Added debug image outputs - Cleaned up including removing GradientPenaltyLoss, custom SpectralNorm - Removed all the custom discriminators		2020-08-02 10:45:24 -06:00
..
archs	Update how attention_maps are created	2020-08-01 20:23:46 -06:00
SPSR_modules	Substantial SPSR mods & fixes	2020-08-02 10:45:24 -06:00
__init__.py	Add SPSR_module	2020-08-01 22:02:54 -06:00
base_model.py
feature_model.py
loss.py	Substantial SPSR mods & fixes	2020-08-02 10:45:24 -06:00
lr_scheduler.py
networks.py
SPSR_model.py	Substantial SPSR mods & fixes	2020-08-02 10:45:24 -06:00
SPSR_networks.py	Add SPSR_module	2020-08-01 22:02:54 -06:00
SR_model.py
SRGAN_model.py