tinygrad/models
George Hotz 810f03dafa
conv3d + unet3d (#772)
* conv3d, needs test

* test passes, padding wrong on unet

* unet3d

* no conv3d on images
2023-05-12 13:54:07 -07:00
..
convnext.py LayerNorm2d for 2 lines 2023-03-20 16:58:43 -07:00
efficientnet.py third try at torch loading (#677) 2023-03-10 19:11:29 -08:00
resnet.py start on mlperf models 2023-05-10 16:30:49 -07:00
transformer.py good changes from llama branch (#671) 2023-03-09 20:51:22 -08:00
unet3d.py conv3d + unet3d (#772) 2023-05-12 13:54:07 -07:00
vit.py good changes from llama branch (#671) 2023-03-09 20:51:22 -08:00