将GitHUB上下载的模块用于自己的模型，一定要注意输入数据的数据类型（dtype）是否满足模块的要求
第一步：数据类型为张量：train_inputs=torch.tensor(train_inputs)；
第二步：数据中的元素类型：（根据自身需要：int、flaot16、float32）
train_inputs=torch.tensor(train_inputs,dtype=torch.float32)
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
