DL-Art-School/codes/utils
James Betker 7629cb0e61 Add FDPL Loss
New loss type that can replace PSNR loss. Works against the frequency domain
and focuses on frequency features loss during hr->lr conversion.
2020-07-30 20:47:57 -06:00
..
__init__.py
colors.py Add FDPL Loss 2020-07-30 20:47:57 -06:00
convert_model.py Add switched discriminator 2020-07-22 20:52:59 -06:00
distill_torchscript.py Misc changes 2020-07-10 09:45:48 -06:00
fdpl_util.py Add FDPL Loss 2020-07-30 20:47:57 -06:00
gpu_mem_track.py Add GPU mem tracing module 2020-06-14 11:02:54 -06:00
numeric_stability.py Add u-net discriminator with feature output 2020-07-16 10:10:09 -06:00
onnx_inference.py Fix temperature equation 2020-07-14 10:17:14 -06:00
util.py