MaxPoolingOp only supports NHWC. ERROR in my benchmark we would use NCHW dataformat

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/memories_sunset/article/details/83039247

基于Tensorflow跑SSD遇到的大坑

  • 网上所使用的SSD代码是基于CPU的,因此其只能够支持NHWC格式的。那么这个应该从哪里修改呢?
    在SSD项目中将DATA_FORMAT进行修改

猜你喜欢

转载自blog.csdn.net/memories_sunset/article/details/83039247