James Betker dffc15184d More ExtensibleTrainer work ... It runs now, just need to debug it to reach performance parity with SRGAN. Sweet.		2020-08-23 17:22:45 -06:00
..
__init__.py	mmsr	2019-08-23 21:42:47 +08:00
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
loss_accumulator.py	More ExtensibleTrainer work	2020-08-23 17:22:45 -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	Introduce (untested) colab mode	2020-06-01 15:09:52 -06:00