George Hotz 7276f8d6bf improve constant folding, detach before moving tensor

2022-07-02 15:29:40 -07:00

..

yolo

Fix yolo param name (#306)

2022-02-10 01:40:24 -07:00

benchmark_train_efficientnet.py

improve constant folding, detach before moving tensor

2022-07-02 15:29:40 -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

support drawing graphs

2022-01-16 10:45:58 -08:00

train_resnet.py

Transfer learning for ResNet (#295)

2022-01-15 23:22:10 -05:00

transformer.py

add ff_dim to transformer

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

vgg7.py

Correct spelling mistakes (#314)

2022-04-05 05:22:18 -07:00

vit.py

support larger ViT models

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

yolov3.py

Correct spelling mistakes (#314)

2022-04-05 05:22:18 -07:00