James Betker 9ccf771629 Fix feature validation, wrong device ... Only shows up in distributed training for some reason.		2020-07-23 10:16:34 -06:00
..
archs	Add switched discriminator	2020-07-22 20:52:59 -06:00
__init__.py
base_model.py
loss.py	Add u-net discriminator with feature output	2020-07-16 10:10:09 -06:00
lr_scheduler.py	Huge set of mods to support progressive generator growth	2020-07-18 14:18:48 -06:00
networks.py	Add switched discriminator	2020-07-22 20:52:59 -06:00
SR_model.py
SRGAN_model.py	Fix feature validation, wrong device	2020-07-23 10:16:34 -06:00