PyTorch——错误集锦
来源:互联网 发布:全国cad应用培训网络 编辑:程序博客网 时间:2024/06/05 06:51
1、KeyError: class ‘torch.cuda.ByteTensor’
解决
关于这个错误网上介绍的并不多,只找到一个解答:ByteTensor not working with F.conv2d?。PyTorch中的大部分操作都是针对FloatTensor和DoubleTensor的。
2、RuntimeError: CUDNN_STATUS_BAD_PARAM
解决
输入的size不正确,卷积层的输入size为(N, C, H, W)。
3、TypeError: max() got an unexpected keyword argument ‘keepdim
原因不明。
解决 torch.max(input, dim)
而不用torch.max(input, dim, keepdim)
4、RuntimeError: getCudnnDataType() not supported for B
调用Module.forward()方法,在计算Conv2d时出现这个错误。
解决
网络的输入必须是类型为Float或者Double或者Half Tensor且必须封装在Variable中。
5、cuda out of memory
在训练经过一段时间后出现out of memory错误,这意味着内存占用在训练过程中不断增加。
原因
loss或者网络的输出不断积累,导致计算图不断扩张。
解决
如果在训练的循环过程中需要用到loss,应该使用loss.data[0]。
阅读全文
0 0
- PyTorch——错误集锦
- PyTorch学习—PyTorch是什么?
- Torch——错误集锦
- PyTorch(一)——数据处理
- Pytorch入门——概述
- Pytorch入门——安装
- Pytorch入门——Tensor
- Pytorch入门——神经网络
- Pytorch入门——安装
- 感知机/Pytorch—笔记
- PyTorch学习3—神经网络
- Android开发——错误集锦
- 【Oracle】——安装Oracle10g错误集锦
- MQ整理—搭建错误集锦
- PyTorch的错误
- [Pytorch]: 错误汇总
- PyTorch之示例——MNIST
- Pytorch入门——Autograd:自动分化
- 怎么学好php
- 流的作用
- 基于JZ2440的NOR FLASH的驱动程序的实现
- 为 CentOS 6.5 配置163 yum 源
- 范数介绍及C++/OpenCV/Eigen的三种实现
- PyTorch——错误集锦
- Android studio安装ButterKnife插件
- 20170522 工作日志
- 低功耗蓝牙学习指南0519
- 网易实习--编程题
- selenium3+python2.7启动Firefox53提示:KeyError: 'sessionId'
- IO流
- 一把钥匙开一把锁
- 第一个文章