Open undertherain opened 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
check if it is same as mlperf specs, that is using ResNet50 as a back-bone CNN
Seems like Resnet34 (as mentioned in slack):
https://github.com/mlperf/training/blob/fb058e3849c25f6c718434e60906ea3b0cb0f67d/single_stage_detector/ssd/ssd300.py#L21
From slack! We will use coco2017 (seems to be the latest)
there's pretrained one from Nvidia on torch hub - but it does not deserialize without GPU convert to CPU and upload somewhere