if you do not mind this loss, you can: 1) explicitly cast your data to float32

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

很明显的错误提示:
这里写图片描述

一开始以为是tensor的类型不对,使用cast函数转换,发现不是,原来是numpy的问题,直接将原变量使用如下进行转换

data = data.astype(float32)
阅读更多
想对作者说点什么?

博主推荐

换一批

没有更多推荐了,返回首页