forked from mrq/DL-Art-School
645d0ca767
- Add filters_mid spec which allows a expansion->squeeze for the transformation layers. - Add scale and bias AFTER the switch - Remove identity transform (models were converging on this) - Move attention image generation and temperature setting into new function which gets called every step with a save path |
||
---|---|---|
.. | ||
archs | ||
__init__.py | ||
base_model.py | ||
loss.py | ||
lr_scheduler.py | ||
networks.py | ||
SR_model.py | ||
SRGAN_model.py |