George Hotz 9faf205601 CIFAR trainer + various bugfixes / improvements (#6146) ... * move cifar into datasets * support for pathlib Tensors, tar_extract, and fetch gunzip * too early for Device.DEFAULT * simpler hlb_cifar + .to(None) is default * new compiler failure, start beautiful_cifar * beautiful cifar runs but is broken * jit train step * cleaner * std_mean, not mean_std * more correct * fast indexing * don't print that * torch load broken * add eval * nicer bar * decoraters are the way to do this * bounds check the load * a few ops * batchnorm bugfix, if track_running_stats is False, use online estimate * full timing * fix fusion * unneeded realize * master tensor		2024-08-20 16:58:46 -07:00
..
conversation_data	…
llm.c	move memory planner to realize (#5937)	2024-08-06 10:41:29 -07:00
mlperf	no load in INITMLPERF (#5957)	2024-08-08 11:28:24 -04:00
openpilot	merge uops with ops (#6111)	2024-08-16 18:17:57 -04:00
other_mnist	…
rl	…
sovits_helpers	…
tinychat	…
vgg7_helpers	…
webgl/yolov8	…
webgpu/stable_diffusion	…
__init__.py	…
beautiful_cartpole.py	…
beautiful_cifar.py	CIFAR trainer + various bugfixes / improvements (#6146)	2024-08-20 16:58:46 -07:00
beautiful_mnist.py	tensor inference (#6156)	2024-08-18 00:19:28 -07:00
beautiful_mnist_multigpu.py	…
coder.py	…
compile_efficientnet.py	…
compile_tensorflow.py	…
conversation.py	…
efficientnet.py	…
gpt2.py	…
handcode_opt.py	merge uops with ops (#6111)	2024-08-16 18:17:57 -04:00
hlb_cifar10.py	add cifar to datasets.py (#6210)	2024-08-20 11:42:49 -07:00
index.html	…
llama.py	shard kvcache (#5830)	2024-07-30 20:29:54 -07:00
llama3.py	faster tinychat (#5993)	2024-08-08 19:16:26 -07:00
mamba.py	…
mask_rcnn.py	…
mixtral.py	…
mnist_gan.py	…
openelm.py	…
sdv2.py	…
sdxl.py	…
sdxl_seed0.png	…
serious_mnist.py	…
simple_conv_bn.py	…
so_vits_svc.py	…
stable_diffusion.py	…
stable_diffusion_seed0.png	…
train_efficientnet.py	…
train_resnet.py	…
transformer.py	…
vgg7.py	…
vit.py	…
vits.py	…
whisper.py	…
yolov3.py	…
yolov8-onnx.py	…
yolov8.py	…