George Hotz 99b6051467 add ff_dim to transformer

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

..

yolo

move yolo into examples/yolo

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

efficientnet.py

move good models to models, add a training step test

2021-06-19 11:24:15 -07: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

move datasets into datasets

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

train_resnet.py

Added ResNet-{18, 34, 50, 101, 152} (#271)

2021-06-21 09:37:24 -07: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

add ff_dim to transformer

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

yolov3.py

move yolo into examples/yolo

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