James Betker 3e3ad7825f Add support for training an EMA network alongside the main networks		2021-06-12 21:01:41 -06:00
..
__init__.py
colors.py
convert_model.py	Codebase cleanup	2020-10-13 20:56:39 -06:00
distributed_checkpont.py
fdpl_util.py
gpu_mem_track.py
kmeans.py	test uresnet playground mods	2021-01-23 13:46:43 -07:00
loss_accumulator.py	More work on SSIM/PSNR approximators	2020-11-03 08:09:58 -07:00
numeric_stability.py	More refactor changes	2020-12-18 09:24:31 -07:00
options.py	Add support for training an EMA network alongside the main networks	2021-06-12 21:01:41 -06:00
util.py	Support gaussian diffusion models	2021-06-02 21:47:32 -06:00
weight_scheduler.py	Codebase cleanup	2020-10-13 20:56:39 -06:00