13 Commits

Author	SHA1	Message	Date
James Betker	68cbbed886	Add some cool diffusion testing scripts	2021-06-16 16:26:36 -06:00
James Betker	5b4f86293f	Add FID evaluator for diffusion models	2021-06-14 09:14:30 -06:00
James Betker	65c474eecf	Various changes to fix testing	2021-06-11 15:31:10 -06:00
James Betker	7c5478bc2c	Formatting issue with gdi	2021-06-06 16:35:37 -06:00
James Betker	692e9c417b	Support diffusion unet	2021-06-06 13:57:22 -06:00
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
James Betker	bf811f80c1	GD mods & fixes ... - Report variational loss separately - Report model prediction from injector - Log these things - Use respacing like guided diffusion	2021-06-04 17:13:16 -06:00
James Betker	6084915af8	Support gaussian diffusion models ... Adds support for GD models, courtesy of some maths from openai. Also: - Fixes requirement for eval{} even when it isn't being used - Adds support for denormalizing an imagenet norm	2021-06-02 21:47:32 -06:00
James Betker	f89ea5f1c6	Mods to support lightweight_gan model	2021-03-02 20:51:48 -07:00
James Betker	784b96c059	Misc options to add support for training stylegan2-rosinality models: ... - Allow image_folder_dataset to normalize inbound images - ExtensibleTrainer can denormalize images on the output path - Support .webp - an output from LSUN - Support logistic GAN divergence loss - Support stylegan2 TF weight extraction for discriminator - New injector that produces latent noise (with separated paths) - Modify FID evaluator to be operable with rosinality-style GANs	2021-02-08 08:09:21 -07:00
James Betker	acf1535b14	Fix for randomresizedcrop injector	2021-01-07 16:31:43 -07:00
James Betker	04961b91cf	Add random-crop injector	2021-01-07 12:14:55 -07:00
James Betker	63cf3d3126	Injector auto-registration ... I love it!	2020-12-29 20:58:02 -07:00