Pytorch
CaoChengtai
这个作者很懒,什么都没留下…
展开
-
RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM
问题描述:在将输入数据送入到网络模型的时候,报错:RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM解决:一开始在网上查找相关问题描述,其中绝大部分说的是显卡中缓存太多,需要清理缓存然后重新启动。但是按照他们介绍的步骤,发现问题并没有得到解决,而且查看显卡状态,显存状态良好。最后在 stackoverflow 上找到的解决方法方法:...原创 2019-12-04 17:24:40 · 19168 阅读 · 10 评论 -
Pytorch 之 MNIST 数据集实现
目录1. 数据集介绍2. 代码2. 读代码(个人喜欢的顺序)2.1. 导入模块部分:2.2. Main 函数:1. 数据集介绍一般而言,MNIST 数据集测试就是机器学习和深度学习当中的"Hello World"工程。几乎是所有的教程都会把它放在最开始的地方。这是因为,这个简单的工程包含了大致的机器学习流程,通过练习这个工程有助于读者加深理解机器学习或者是深度学习的大致流程。MNIST(Mi...原创 2019-07-14 10:43:35 · 24310 阅读 · 13 评论 -
Pytorch--Tensor, Numpy--Array,Python--List 相互之间的转换。
1.1 List --> Arrary: np.array(List 变量)a = [1, 2, 3, 4]b = np.array(a)1.2 Arrary --> List: Array 变量.tolist()a = [1, 2, 3, 4]b = np.array(a)c = b.tolist()2.1 List --> Tensor: torch.T...原创 2019-08-13 17:02:34 · 15719 阅读 · 0 评论 -
RuntimeError: Expected object of scalar type Double but got scalar type Float for argument #2 'mat2'
问题:RuntimeError: Expected object of scalar type Double but got scalar type Float for argument #2 ‘mat2’问题代码段:embedding1 = network.forward(train_support)解决方法: train_support = train_support.clone()...原创 2019-08-23 09:21:19 · 2732 阅读 · 0 评论