James Betker 80d4404367 A few fixes: ... - Output better prediction of xstart from eps - Support LossAwareSampler - Support AdamW		2021-06-05 13:40:32 -06:00
..
.idea	IDEA update	2020-05-19 09:35:26 -06:00
data	Allow validation for ce	2021-06-04 21:21:04 -06:00
models	A few fixes:	2021-06-05 13:40:32 -06:00
scripts	Fixes and mods to support training classifiers on imagenet	2021-06-01 17:25:24 -06:00
trainer	A few fixes:	2021-06-05 13:40:32 -06:00
utils	Support gaussian diffusion models	2021-06-02 21:47:32 -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	misc adjustments for stylegan	2021-04-21 18:14:17 -06:00
train.py	Fix wandb import issue	2021-06-04 23:27:15 -06:00