- 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
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.