DL-Art-School/codes/models/SPSR_modules
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
..
__init__.py Add SPSR_module 2020-08-01 22:02:54 -06:00
architecture.py Substantial SPSR mods & fixes 2020-08-02 10:45:24 -06:00
block.py Substantial SPSR mods & fixes 2020-08-02 10:45:24 -06:00
loss.py Substantial SPSR mods & fixes 2020-08-02 10:45:24 -06:00
sampler.py Add SPSR_module 2020-08-01 22:02:54 -06:00