tinygrad/models
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