12 Commits

Author	SHA1	Message	Date
George Hotz	1240c12ac5	download cifar to datasets dir	2023-03-29 12:25:42 +04:00
George Hotz	623fb1ef28	do test_conv_with_bn test	2023-03-19 23:53:56 -07:00
George Hotz	ba3bf5bdf7	cifar stops learning	2023-02-11 17:21:42 -08:00
George Hotz	f7291f6ca3	fixes big KOPT, breaks opencl (#505) ... * fixes big KOPT, breaks opencl * fix optimizer * KernelCache * oops, broke batchnorm * hack to fix it * fix llvm, less hacky gpu * disable the cache * cache just breaks things	2023-02-05 10:46:17 -08:00
George Hotz	c1a769b68b	fix bug in gpu copy out	2023-01-30 16:51:28 -08:00
George Hotz	cccfea4b25	factor out KOPT code	2023-01-30 13:13:55 -08:00
George Hotz	a0c0239ff1	fix mnist load from other dirs	2022-09-25 12:50:28 -04:00
George Hotz	c1d5af8b0c	stable diffusion cleanups	2022-09-05 18:34:13 -07:00
George Hotz	114728d363	torch bs	2022-09-03 11:57:23 -07:00
George Hotz	2cae2dfa07	don't crash the dataloader for imagenet	2022-01-16 08:41:26 -08:00
George Hotz	d1e082e0ef	add imagenet training support	2022-01-15 23:16:38 -08:00
George Hotz	b0f14b4af8	move datasets into datasets	2021-10-30 19:55:50 -07:00