tinygrad/models
JudeDavis1 f3168ee69b
default transformer dropout to 0 (#828)
* default mha dropout to 0

* simplify assert

* reform

* default to 0.1
2023-05-29 08:06:16 -07:00
..
bert.py Add mlperf bert model (#803) 2023-05-27 14:53:32 -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 add retinanet with resnet backbone (#813) 2023-05-28 20:20:16 -07:00
retinanet.py add retinanet with resnet backbone (#813) 2023-05-28 20:20:16 -07:00
rnnt.py Promote Embedding to nn (#798) 2023-05-25 18:39:45 -07:00
transformer.py default transformer dropout to 0 (#828) 2023-05-29 08:06:16 -07:00
unet3d.py Add MLPerf UNet3D model (#775) 2023-05-28 20:38:19 -07:00
vit.py good changes from llama branch (#671) 2023-03-09 20:51:22 -08:00