5f2c722a10
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. |
||
|---|---|---|
| .. | ||
| .idea | ||
| data | ||
| data_scripts | ||
| metrics | ||
| models | ||
| options | ||
| scripts | ||
| temp | ||
| utils | ||
| distill_torchscript.py | ||
| onnx_inference.py | ||
| process_video.py | ||
| recover_tensorboard_log.py | ||
| requirements.txt | ||
| run_scripts.sh | ||
| test.py | ||
| train.py | ||