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
SPSR_modules	Substantial SPSR mods & fixes	2020-08-02 10:45:24 -06:00
__init__.py
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
SR_model.py
SRGAN_model.py