James Betker bba283776c Enable find_unused_parameters for DistributedDataParallel ... attention_norm has some parameters which are not used to compute grad, which is causing failures in the distributed case.		2020-07-23 09:08:13 -06:00
..
archs	Add switched discriminator	2020-07-22 20:52:59 -06:00
__init__.py
base_model.py
loss.py
lr_scheduler.py
networks.py	Add switched discriminator	2020-07-22 20:52:59 -06:00
SR_model.py
SRGAN_model.py	Enable find_unused_parameters for DistributedDataParallel	2020-07-23 09:08:13 -06:00