tinygrad/models
2023-07-01 12:04:19 -07:00
..
bert.py remove unused imports in models (#1088) 2023-07-01 12:04:19 -07:00
convnext.py LayerNorm2d for 2 lines 2023-03-20 16:58:43 -07:00
efficientnet.py remove unused imports in models (#1088) 2023-07-01 12:04:19 -07:00
mask_rcnn.py MaskRCNN Inference (#884) 2023-06-25 15:37:51 -07:00
resnet.py remove unused imports in models (#1088) 2023-07-01 12:04:19 -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