@jiazhihao hi,I met a problem when I try to optimize inception_v3 model with TASO.Could you please show me how to deal with it and how to make all the operators of inception_v3 supported by TASO?thx
step:
1.convert inception_v3 model from .pb to .onnx (use the tf2onnx)
2.load and optimize inception_v3.onnx with TASO
this occured when I load inception_v3.onnx into TASO
(vir-taso) [root@centos /taso-master/taso/examples/myexamples]# python inceptionV3_taso_opt.py
Load onnx model...
0 Add
cuDNN does noot suppoort zero stride for broadcast
Consider switch to other library for broadcastable operators.
python: /taso-master/taso/src/cudnn/cuda_helper.cu:94: void helperSetBroadcastableTensorDescriptor(const taso::Tensor&, const taso::Tensor&, cudnnTensorDescriptor_t): Assertion `false' failed.
Aborted
@jiazhihao hi,I met a problem when I try to optimize inception_v3 model with TASO.Could you please show me how to deal with it and how to make all the operators of inception_v3 supported by TASO?thx step: 1.convert inception_v3 model from .pb to .onnx (use the tf2onnx) 2.load and optimize inception_v3.onnx with TASO this occured when I load inception_v3.onnx into TASO
(vir-taso) [root@centos /taso-master/taso/examples/myexamples]# python inceptionV3_taso_opt.py Load onnx model... 0 Add cuDNN does noot suppoort zero stride for broadcast Consider switch to other library for broadcastable operators. python: /taso-master/taso/src/cudnn/cuda_helper.cu:94: void helperSetBroadcastableTensorDescriptor(const taso::Tensor&, const taso::Tensor&, cudnnTensorDescriptor_t): Assertion `false' failed. Aborted
Some useful infomation inception_v3 model: https://github.com/dmlc/web-data/blob/master/tensorflow/models/InceptionV3/inception_v3_2016_08_28_frozen-with_shapes.pb GPU:NVIDIA Tesla P100 PCIe 16GB
CUDA Version 9.0.176 / CUDNN Version 7.4.1