- 博客(4)
- 收藏
- 关注
RSS订阅原创 用PyTorch进行训练时报错RuntimeError: Function AddBackward0 returned an invalid gradient at index 1
在用Pytorch进行训练的时候，进行loss.backward()一步时可能出现如下的问题：会出现以下错误：RuntimeError: Function AddBackward0 returned an invalid gradient at index 1 - expected type TensorOptions(dtype=float, device=cuda:0, layout=Strided, requires_grad=false) but got TensorOptions(dtype=f
2021-06-18 17:47:10
5863
原创 训练时出现的显存错误CUDA error: out of memory
RuntimeError: CUDA error: out of memory出现这种错误一般分两种情况：第一种情况是你的显卡的显存确实不够，解决方法就是改动 bathsize 这些超参数试试，或者氪金买卡；第二种情况是显卡的性能能满足，但是被其他的进程占用了，所以可分配的显存不足，解决方法就是结束这些占用显卡的进程从而使现存得到释放。具体地：在Linux中，在终端输入nvidia-smi查看占用显存的PID进程号，然后输入kill -9 -PID在Windows中，直
2021-06-18 17:16:47
8537
原创 YOLOv1的损失函数和关键概念详解
Loss function详解：在loss function中，前面两行表示localization error(即坐标误差)，第一行是box中心坐标(x,y)的预测，第二行为宽和高的预测。这里注意用宽和高的开根号代替原来的宽和高，这样做主要是因为相同的宽和高误差对于小的目标精度影响比大的目标要大。举个例子，原来w=10，h=20，预测出来w=8，h=22，跟原来w=3，h=5，预测出来w1，h=7相比，其实前者的误差要比后者小，但是如果不加开根号，那么损失都是一样：4+4=8，但是加上根号后，变成0..
2020-07-15 15:54:02
4144
5
原创 虚拟环境安装pytorch
虚拟环境安装pytorch第一步：打开Anaconda的Anaconda Prompt（默认已经装好Anaconda）创建虚拟环境：conda create -n pytorch python=3.7按y同意安装相关的包第二步：安装pytorch:分为GPU版本和CPU版本，打开Pytorch官网CPU版本：cuda选项选none,然后复制最下方的安装命令：conda install pytorch torchvision cpuonly -c pytorch在创建的pytorch环境
2020-06-02 10:05:28
5034
4
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人