Kunwar Raj Singh 5d3310ce56 MaskRCNN Inference (#884) ... * MaskRCNN weights loading * backbone maybe works * backbone works, but resnet body atol 1e-3 * RPN Call, but veryy wrong output * fixed topk * RPN maybe works, not sure about nms * Fix cursed modules * add back editorconfig * Full call, wrong output * Full call works * fix mask * use NMS from retinanet * Removing extra funcs * refactor * readable * Add example to run model * remove filter * Fix split, batched inference is worse * Fix image sizes * Matching reference * merge master * add filter on top detections * cuda backend fixed * add model eval and spec * convert images to rgb * fix eval * simplify examples code * remove extra code * meshgrid using tinygrad * removing numpy * roi align, floor, ceil * remove numpy from level_mapper * remove numpy from pooler * Revert "Merge branch 'master' of github.com:kunwar31/tinygrad into mrcnn-inference" This reverts commit 4b95a3cb49, reversing changes made to 98f2b1fa2e. * roi align gather * fix master merge * revert to old floor, ceil as ints present in domain * use log2 op * fix indexes * weird bug with ints and gpu * weird bug with ints and gpu * refactors, add env var for gather * floor with contiguous, where * refactor topk, sort * remove staticmethod * refactor stride * remove log2 mlop * realize -> contiguous * refactor forward * remove num_classes, stride_in_1x1 from state * refactor forward * refactoring * flake8 * removing numpy in anchor gen, use numpy for gather, nonzero, optimize topk * keep using tinygrad for smaller gathers * fix empty tensors * comms * move from tensor.py * resnet test passing * add coco dataset back * fix spaces * add test for log2 * no need to create Tensors * no need to create Tensors --------- Co-authored-by: Kunwar Raj Singh <kunwar31@pop-os.localdomain>		2023-06-25 15:37:51 -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	Refactor LoadOps (#910)	2023-06-03 09:40:43 -07:00
mask_rcnn.py	MaskRCNN Inference (#884)	2023-06-25 15:37:51 -07:00
resnet.py	MaskRCNN Inference (#884)	2023-06-25 15:37:51 -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