James Betker a0ef07ddb8 Create unet_latent_guide.py		2021-06-25 11:25:14 -06:00
..
.idea	IDEA update	2020-05-19 09:35:26 -06:00
data	Try to make diffusion fid more deterministic	2021-06-14 09:27:43 -06:00
models	Create unet_latent_guide.py	2021-06-25 11:25:14 -06:00
scripts	Add some cool diffusion testing scripts	2021-06-16 16:26:36 -06:00
trainer	Fix a bug where non-rank-0 is computing FID before all images are saved.	2021-06-16 16:27:09 -06:00
utils	Add support for training an EMA network alongside the main networks	2021-06-12 21:01:41 -06:00
multi_modal_train.py	More adjustments to support distributed training with teco & on multi_modal_train	2020-10-27 20:58:03 -06:00
process_video.py	misc	2021-01-23 13:45:17 -07:00
requirements.txt	Log eval to wandb	2021-06-04 23:23:20 -06:00
test_image_patch_classifier.py	More refactoring	2020-12-18 09:18:34 -07:00
test.py	Add FID evaluator for diffusion models	2021-06-14 09:14:30 -06:00
train.py	Another fix	2021-06-14 09:51:44 -06:00