tinygrad/models
George Hotz b1206bcb18
third try at torch loading (#677)
* third try at torch loading

* numpy fixed

* fix enet compile

* load_single_weight supports empty weights

* oops, CPU wasn't the default

* so many bugs
2023-03-10 19:11:29 -08:00
..
convnext.py ConvNeXt + extras (#657) 2023-03-06 22:10:56 -08:00
efficientnet.py third try at torch loading (#677) 2023-03-10 19:11:29 -08:00
resnet.py rename log_softmax, support dim, fix onnx Softmax 2023-02-24 10:11:24 -08:00
transformer.py good changes from llama branch (#671) 2023-03-09 20:51:22 -08:00
vit.py good changes from llama branch (#671) 2023-03-09 20:51:22 -08:00