当前位置:网站首页>数据切片问题
数据切片问题
2022-08-10 03:28:00 【小乐快乐】
在cpu的mindspore r.15跑model_zoo官方faster rcnn 的ops Squeeze 算子或切片出错
【操作步骤&问题现象】
1、Tensor的 size为(2,128,4)
2、在rpn中进行 gt_bboxes_i = self.squeeze(gt_bboxes[i:i + 1:1, ::]) 或 gt_bboxes[i,::]报错 但自己test生成tensor切片不报错
自己写的:
【截图信息】
【日志信息】(可选,上传日志内容或者附件)
Traceback (most recent call last):
File "E:/Mindspore1.1/mindspore/model_zoo/official/cv/faster_rcnn/src/FasterRcnn/faster_rcnn_r50.py", line 514, in <module>
x = network(x,img_metas,box,label,vaild_num)
File "F:\py375\lib\site-packages\mindspore\nn\cell.py", line 433, in __call__
raise err
File "F:\py375\lib\site-packages\mindspore\nn\cell.py", line 430, in __call__
output = self.run_construct(cast_inputs, kwargs)
File "F:\py375\lib\site-packages\mindspore\nn\cell.py", line 352, in run_construct
output = self.construct(*cast_inputs, **kwargs)
File "E:/Mindspore1.1/mindspore/model_zoo/official/cv/faster_rcnn/src/FasterRcnn/faster_rcnn_r50.py", line 219, in construct
gt_valids)
File "F:\py375\lib\site-packages\mindspore\nn\cell.py", line 433, in __call__
raise err
File "F:\py375\lib\site-packages\mindspore\nn\cell.py", line 430, in __call__
output = self.run_construct(cast_inputs, kwargs)
File "F:\py375\lib\site-packages\mindspore\nn\cell.py", line 352, in run_construct
output = self.construct(*cast_inputs, **kwargs)
File "E:\Mindspore1.1\mindspore\model_zoo\official\cv\faster_rcnn\src\FasterRcnn\rpn.py", line 282, in construct
gt_labels_i = self.squeeze(gt_labels[i:i + 1:1, ::])
File "F:\py375\lib\site-packages\mindspore\common\tensor.py", line 248, in __getitem__
out = tensor_operator_registry.get('__getitem__')(self, index)
File "F:\py375\lib\site-packages\mindspore\ops\composite\multitype_ops\_compile_utils.py", line 38, in _tensor_getitem
return tensor_index_by_tuple(self, index)
File "F:\py375\lib\site-packages\mindspore\ops\composite\multitype_ops\_compile_utils.py", line 371, in tensor_index_by_tuple
return _tensor_getitem_by_tuple_slice(data, tuple_index)
File "F:\py375\lib\site-packages\mindspore\ops\composite\multitype_ops\_compile_utils.py", line 405, in _tensor_getitem_by_tuple_slice
return P.StridedSlice(0, 0, 0, 0, shrink_axis_mask)(data, begin_strides, end_strides, step_strides)
File "F:\py375\lib\site-packages\mindspore\ops\primitive.py", line 247, in __call__
return _run_op(self, self.name, args)
File "F:\py375\lib\site-packages\mindspore\common\api.py", line 77, in wrapper
results = fn(*arg, **kwargs)
File "F:\py375\lib\site-packages\mindspore\ops\primitive.py", line 682, in _run_op
output = real_run_op(obj, op_name, args)
TypeError: mindspore\ccsrc\runtime\device\cpu\kernel_select_cpu.cc:219 KernelNotSupportException] Operator[StridedSlice] input(kNumberTypeUInt8) output(kNumberTypeUInt8) is not support. This error means the current input type is not supported, please refer to the MindSpore doc for supported types.
Trace:
针对您的问题,我这边看了下,是cpu上StridedSlice算子暂不支持切分uint8类型的Tensor。请问您这里的gt_lables是一个uint8类型的Tensor吗? 用Ascend910跑通了
边栏推荐
猜你喜欢
随机推荐
matlab simulink response spectrum calculation
数据库设计中反映用户对数据要求的模式叫什么
电话自动拨号在电脑上自动拨打
The same is a primary test, why does he pay 5,000 yuan more than me?
Dynamic Web Development Fundamentals
Flink学习15:Flink自定义数据源
动态网页开发基础
小程序wxs
program internationalization
cuda——nms
goland json.Marshal导致&变成\u0026
Flink CDC 2.0及其他数据同步工具对比
金融财经翻译的行业前景如何
学习总结week4_1json
TCP协议之《TCP_CORK选项》
驱动程序开发:按键中断之异步通知
YAPI使用
TCP协议之《Out-Of-Window报文限速》
微信小程序相互跳转如何携带参数
golang:base64编解码(转)