Sohaib 65d09031f2 add retinanet with resnet backbone (#813) ...

* add retinanet with resnet backbone

* adds resnext to support loading retinanet pretrained on openimages
* object detection post processing with numpy
* data is downloaded and converted to coco format with fiftyone
* data loading and mAP evaluation with pycocotools

* remove fiftyone dep

* * eval freq

* fix model timing

* del jit for last batch

* faster accumulate

2023-05-28 20:20: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

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