James Betker be7982b9ae Add skip heads to switcher ... These pass through the input so that it can be selected by the attention mechanism.		2020-06-14 12:46:54 -06:00
..
archs	Add skip heads to switcher	2020-06-14 12:46:54 -06:00
__init__.py
base_model.py	Introduce (untested) colab mode	2020-06-01 15:09:52 -06:00
loss.py
lr_scheduler.py	Enable forced learning rates	2020-06-07 16:56:05 -06:00
networks.py	Add skip heads to switcher	2020-06-14 12:46:54 -06:00
SR_model.py	Allow validating in batches, remove val size limit	2020-06-02 08:41:22 -06:00
SRGAN_model.py	Add GPU mem tracing module	2020-06-14 11:02:54 -06:00