1
0
Fork 0
Commit graph

8 commits

Author SHA1 Message Date
George Hotz 2305a5347b test_onnx works with enet also 2022-06-11 14:30:26 -07:00
George Hotz 85d17a2acd running resnet onnx 2022-06-11 13:17:15 -07:00
George Hotz 70561f3d90 way over the line limit 2022-06-08 09:36:31 -07:00
George Hotz 3dac8fa728 this fix the gc 2022-06-05 17:16:40 -07:00
George Hotz 0ee21ba115 add ViT test and car 2022-06-05 17:12:43 -07:00
George Hotz 46bbbcf7f0 model touchups 2021-11-30 11:13:34 -05:00
George Hotz 7472a7ebe2 not forcing 3.9 for a stupid type 2021-10-30 16:52:40 -07:00
Sebastian Kreft 8113eec4cf
feat: add efficientnet test (#285)
Simple test using the Chicken example from https://upload.wikimedia.org/wikipedia/commons/4/41/Chicken.jpg and the image preprocessing from example/efficientnet.py

Note that EfficientNet loads the weights from the internet so running the tests may be slow the first time. We could speed up the tests by caching the /tmp folder.

Fixes #234
2021-10-30 15:53:51 -07:00