使用sklearn.neighbors.NearestNeighbors.kneighbors()计算tensor计算速度非常缓慢

我在使用mindspore复现网络的时候,使用了sklearn.neighbors.NearestNeighbors.kneighbors()的算子来计算tensor,数据的输入类型如图所示

计算速度非常慢,大大影响了网络的训练速度,请问该如何解决

 注意到你图片中数据类型是fp32请问你的执行是在ascend上执行如果是的话ascend fp32支持度不高会导致执行速度比较慢;https://www.mindspore.cn/tutorials/experts/zh-CN/r1.8/others/mixed_precision.html?highlight=æ··å精度   可以参照上述链接转用混精进行运行

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值