- 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 |
||
---|---|---|
.. | ||
__init__.py | ||
architecture.py | ||
block.py | ||
loss.py | ||
sampler.py |