22 Commits

Include renames

Author	SHA1	Message	Date
Phil Wang	0b8cdb4c8b	remove outdated apex in favor of native pytorch AMP	2022-04-13 08:59:18 -07:00
Phil Wang	91cff45939	replace resnets with convnext blocks	2022-01-25 09:02:45 -08:00
Phil Wang	ae42f48f6a	prepare so that unet can work with a channel of one, and also make it so image size is hard coded in diffusion class. preparing for training on protein distograms	2021-06-11 14:06:39 -07:00
Phil Wang	2082046888	set higher num train steps, so non-practitioners do not think it is completed	2020-10-11 13:49:49 -07:00
Phil Wang	3c5b7e2d56	update readme	2020-10-10 10:37:44 -07:00
Phil Wang	ff451f697e	update with new and improved cosine noise scheduler	2020-10-09 21:21:02 -07:00
Phil Wang	3d96532c60	update citations in preparation to add improvements from a iclr 2021 paper	2020-10-05 17:15:28 -07:00
Phil Wang	8e5fb17063	add badge	2020-09-14 13:38:06 -07:00
Phil Wang	4bf28914bc	allow for mixed precision training with fp16 flag	2020-09-08 17:26:23 -07:00
Phil Wang	1307b3115d	add exponential moving average of model, also allow passing in custom noise schedule	2020-09-08 14:58:09 -07:00
Phil Wang	81fb2a0386	add sample	2020-09-08 10:17:48 -07:00
Phil Wang	d70fb08f8a	small helper fn to make sampling more clear	2020-09-07 16:40:36 -07:00
Phil Wang	1f1e42e9f9	update readme	2020-09-06 22:35:37 -07:00
Phil Wang	e1800c1a8d	remove wip, seems to be working	2020-09-06 14:26:23 -07:00
Phil Wang	11f27032ba	offer training class to easily train model off an image directory	2020-09-06 14:22:44 -07:00
Phil Wang	d8472a6220	update readme	2020-09-06 03:44:25 -07:00
Phil Wang	d59d8b05f6	fix a small bug	2020-09-06 01:50:59 -07:00
Phil Wang	365ce0c335	update readme	2020-09-06 01:44:05 -07:00
Phil Wang	0cc4aecfd5	first alpha release	2020-09-06 01:40:57 -07:00
Phil Wang	df94747c0c	Update README.md	2020-08-25 19:24:13 -07:00
Phil Wang	f94876c6fb	Update README.md	2020-08-25 19:23:06 -07:00
Phil Wang	602421c714	Initial commit	2020-08-25 19:22:11 -07:00