fb595e72a4
Need to be able to train 2 nets in one step: the backbone will be entirely separate with its own optimizer (for an extremely low LR). This functionality was already present, just not implemented correctly. |
||
---|---|---|
.. | ||
archs | ||
steps | ||
__init__.py | ||
base_model.py | ||
ExtensibleTrainer.py | ||
feature_model.py | ||
loss.py | ||
lr_scheduler.py | ||
networks.py | ||
novograd.py | ||
SR_model.py | ||
SRGAN_model.py |