17 Commits

Include renames

Author	SHA1	Message	Date
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