James Betker f2977d360c Allow attention_dim in channel attention to be specified, add converter		2021-10-05 17:29:38 -06:00
..
.idea	IDEA update	2020-05-19 09:35:26 -06:00
data	Alterations to diffusion_dvae so it can be used directly on spectrograms	2021-09-23 15:56:25 -06:00
models	Allow attention_dim in channel attention to be specified, add converter	2021-10-05 17:29:38 -06:00
scripts	More cleanup	2021-09-29 14:24:49 -06:00
trainer	More cleanup	2021-09-29 14:24:49 -06:00
utils	Clean up codebase	2021-09-29 09:21:44 -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	Fix annoying warning and add to requirements	2021-08-11 17:32:06 -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	Add dvae with channel attention	2021-10-03 10:52:01 -06:00
use_discriminator_as_filter.py	Various mods to support better jpeg image filtering	2021-06-25 13:16:15 -06:00