6 Commits

Author	SHA1	Message	Date
Haydn Jones	a3a247e6ec	Fixed disabling of gradients in the torch code	2022-06-28 17:57:44 -06:00
Brett Kuprel	9d6b6dcc92	previous commit broke flax model, fixed now	2022-06-28 12:54:58 -04:00
Brett Kuprel	5aa6fe49bf	use cuda if available	2022-06-28 12:47:11 -04:00
Brett Kuprel	8544f59576	use cuda if available	2022-06-28 12:38:31 -04:00
Brett Kuprel	aef24ea157	torch.no_grad(), cleanup	2022-06-28 12:16:44 -04:00
Brett Kuprel	e7001f063c	simplified	2022-06-27 15:46:04 -04:00