mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity
c8da78966b
DL-Art-School/codes/models
History
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 Update how attention_maps are created 2020-08-01 20:23:46 -06:00
SPSR_modules Substantial SPSR mods & fixes 2020-08-02 10:45:24 -06:00
__init__.py Add SPSR_module 2020-08-01 22:02:54 -06:00
base_model.py Add feature_model for training custom feature nets 2020-07-31 11:20:39 -06:00
feature_model.py Add feature_model for training custom feature nets 2020-07-31 11:20:39 -06:00
loss.py Substantial SPSR mods & fixes 2020-08-02 10:45:24 -06:00
lr_scheduler.py Allow multi_step_lr_scheduler to load a new LR schedule when restoring state 2020-07-31 11:21:11 -06:00
networks.py Enable disjoint feature networks 2020-07-31 16:29:47 -06:00
SPSR_model.py Substantial SPSR mods & fixes 2020-08-02 10:45:24 -06:00
SPSR_networks.py Add SPSR_module 2020-08-01 22:02:54 -06:00
SR_model.py Allow validating in batches, remove val size limit 2020-06-02 08:41:22 -06:00
SRGAN_model.py More torch 1.6 fixes 2020-07-31 17:03:20 -06:00