本文探讨了在训练神经网络模型过程中遇到的target数据类型问题，通过案例分析展示了如何从`numpy.array`转换到`torch.tensor`并指定设备为CPU，以避免CUDA运行时错误。博主分享了三种有效解决方案，帮助读者解决类似困扰。
问题描述：
训练模型的过程中，出现
这个问题出现在general.py 中，源码为
return np.array(targets)
按照网上的方法，改为
return np.array(targets.cpu())
没有解决：提示下面错误
后来看到一份博客，也遇到类似的问题
解决方法：
return np.array(torch.tensor(targets, device='cpu'))
扫一扫
11-23
991
991
03-11
2152
2152
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
