geohot 2e56a4793e rename log_softmax, support dim, fix onnx Softmax

2023-02-24 10:11:24 -08:00

..

efficientnet.py

BatchNorm2D -> BatchNorm2d (#558)

2023-02-16 12:31:49 -08:00

resnet.py

rename log_softmax, support dim, fix onnx Softmax

2023-02-24 10:11:24 -08:00

transformer.py

rename log_softmax, support dim, fix onnx Softmax

2023-02-24 10:11:24 -08:00

vit.py

support larger ViT models

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