c8da78966b
- 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 |
||
---|---|---|
.. | ||
.idea | ||
data | ||
data_scripts | ||
metrics | ||
models | ||
options | ||
scripts | ||
temp | ||
utils | ||
process_video.py | ||
recover_tensorboard_log.py | ||
requirements.txt | ||
run_scripts.sh | ||
test.py | ||
train.py |