James Betker 07168ecfb4 Enable vqvae to use a switched_conv variant		2021-01-09 20:53:14 -07:00
..
__init__.py
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	Enable vqvae to use a switched_conv variant	2021-01-09 20:53:14 -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	Get rid of CUDA_VISIBLE_DEVICES	2020-12-31 10:13:58 -07:00
util.py	Add sr_fid evaluator	2020-12-30 20:18:58 -07:00
weight_scheduler.py	Codebase cleanup	2020-10-13 20:56:39 -06:00