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