DL-Art-School/codes/models
James Betker 5f2c722a10 SRG2 revival
Big update to SRG2 architecture to pull in a lot of things that have been learned:
- Use group norm instead of batch norm
- Initialize the weights on the transformations low like is done in RRDB rather than using the scalar. Models live or die by their early stages, and this ones early stage is pretty weak
- Transform multiplexer to use u-net like architecture.
- Just use one set of configuration variables instead of a list - flat networks performed fine in this regard.
2020-07-09 17:34:51 -06:00
..
archs SRG2 revival 2020-07-09 17:34:51 -06:00
__init__.py Implement downsample GAN 2020-04-24 00:00:46 -06:00
base_model.py Fix inverse temperature curve logic and add upsample factor 2020-06-19 09:18:30 -06:00
loss.py Fix up pixgan loss and pixdisc 2020-07-08 21:27:48 -06:00
lr_scheduler.py Enable forced learning rates 2020-06-07 16:56:05 -06:00
networks.py SRG2 revival 2020-07-09 17:34:51 -06:00
SR_model.py Allow validating in batches, remove val size limit 2020-06-02 08:41:22 -06:00
SRGAN_model.py More fixes... 2020-07-08 22:07:09 -06:00