George Hotz 4eabe677ed fixup resnet

2022-01-15 20:21:02 -08:00

..

yolo

move yolo into examples/yolo

2021-10-30 19:46:00 -07:00

efficientnet.py

model touchups

2021-11-30 11:13:34 -05:00

mnist_gan.py

fixed example broken after gpu refactor (#238)

2021-01-05 07:41:54 -08:00

serious_mnist.py

move datasets into datasets

2021-10-30 19:55:50 -07:00

train_efficientnet.py

training is False by default

2022-01-15 19:57:41 -08:00

train_resnet.py

fixup resnet

2022-01-15 20:21:02 -08:00

transformer.py

add ff_dim to transformer

2021-11-29 12:40:52 -05:00

vgg7.py

move yolo into examples/yolo

2021-10-30 19:46:00 -07:00

vit.py

support larger ViT models

2021-12-12 10:45:10 -08:00

yolov3.py

move yolo into examples/yolo

2021-10-30 19:46:00 -07:00