DL-Art-School/codes/utils
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
..
__init__.py mmsr 2019-08-23 21:42:47 +08:00
gpu_mem_track.py Add GPU mem tracing module 2020-06-14 11:02:54 -06:00
numeric_stability.py SRG2 revival 2020-07-09 17:34:51 -06:00
util.py Introduce (untested) colab mode 2020-06-01 15:09:52 -06:00