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 | ||