In the official web site, the format of the link of SSD-ResNet34 model is "NCHW". This format is usually for GPU. It means that we can not use it on TensorFlow.
I found a link after searching google. It seems that the format is "NHWC". I used this mode to run MLPerf v0.5. I got the result finally.
The link is shown below.
https://zenodo.org/record/3246481#.XvmdbCgzaUk
I used these below parameters to run MLPerf.
./run_local.sh tf ssd-resnet34 cpu --data-format NHWC
Although I got the result, there was errors of "only size-1 arrays can be converted to Python scalars". I will check it later.
Reference
https://github.com/mlperf/inference/issues/219
https://github.com/mlperf/inference/issues/229
沒有留言:
張貼留言