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