James Betker f78ce9d924 Get diffusion_dvae ready for prime time!		2021-09-16 22:43:10 -06:00
..
__init__.py	mmsr	2019-08-23 21:42:47 +08:00
audio_resampler.py	Stop dataset - attempt #2	2021-08-18 18:29:38 -06:00
colors.py	Add FDPL Loss	2020-07-30 20:47:57 -06:00
convert_model.py	Codebase cleanup	2020-10-13 20:56:39 -06:00
distributed_checkpont.py	Add distributed_checkpoint for more efficient checkpoints	2020-10-06 20:38:38 -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
kmeans.py	test uresnet playground mods	2021-01-23 13:46:43 -07:00
loss_accumulator.py	Add "pure" evaluator	2021-08-09 14:58:35 -06: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	Get diffusion_dvae ready for prime time!	2021-09-16 22:43:10 -06:00
weight_scheduler.py	Codebase cleanup	2020-10-13 20:56:39 -06:00