DL-Art-School/codes/models
James Betker eb11a08d1c Enable disjoint feature networks
This is done by pre-training a feature net that predicts the features
of HR images from LR images. Then use the original feature network
and this new one in tandem to work only on LR/Gen images.
2020-07-31 16:29:47 -06:00
..
archs Enable disjoint feature networks 2020-07-31 16:29:47 -06:00
__init__.py Add feature_model for training custom feature nets 2020-07-31 11:20:39 -06:00
base_model.py Add feature_model for training custom feature nets 2020-07-31 11:20:39 -06:00
feature_model.py Add feature_model for training custom feature nets 2020-07-31 11:20:39 -06:00
loss.py Add FDPL Loss 2020-07-30 20:47:57 -06:00
lr_scheduler.py Allow multi_step_lr_scheduler to load a new LR schedule when restoring state 2020-07-31 11:21:11 -06:00
networks.py Enable disjoint feature networks 2020-07-31 16:29:47 -06:00
SR_model.py
SRGAN_model.py Enable disjoint feature networks 2020-07-31 16:29:47 -06:00