Commit Graph

6 Commits

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