undertherain / benchmarker

modular framework for [not only] deep learning performance benchmarking
http://blackbird.pw/performance
Mozilla Public License 2.0
9 stars 5 forks source link

add SSD model #33

Open undertherain opened 4 years ago

undertherain commented 4 years ago

there's pretrained one from Nvidia on torch hub - but it does not deserialize without GPU convert to CPU and upload somewhere

undertherain commented 4 years ago

check if it is same as mlperf specs, that is using ResNet50 as a back-bone CNN

vatai commented 4 years ago

Seems like Resnet34 (as mentioned in slack):

https://github.com/mlperf/training/blob/fb058e3849c25f6c718434e60906ea3b0cb0f67d/single_stage_detector/ssd/ssd300.py#L21

vatai commented 4 years ago

From slack! We will use coco2017 (seems to be the latest)