bba283776c
attention_norm has some parameters which are not used to compute grad, which is causing failures in the distributed case. |
||
---|---|---|
.. | ||
archs | ||
__init__.py | ||
base_model.py | ||
loss.py | ||
lr_scheduler.py | ||
networks.py | ||
SR_model.py | ||
SRGAN_model.py |