pytorch
文章平均质量分 67
pytorch深度学习框架
庸中佼佼
这个作者很懒,什么都没留下…
展开
-
mmdetection使用记录
环境配置服务器将整个项目git过来,在环境中依次安装项目所需库conda install pytorch torch 1.8.0 cudatoolkit......pip install openmimmim install mmdetpip install -r requirements/build.txtpip install -v -e . # or "python setup.py develop"需重新创建两个文件夹,checkpoints用于存放模型文件(从readme中进原创 2022-02-19 10:14:41 · 1218 阅读 · 0 评论 -
autograd 自动求导
情况1 out是y的平均值,是一个标量,可以直接得出其对x的梯度(对xix_ixi的梯度之和)import torchx=torch.tensor([[12,6],[31,0.9]],requires_grad=True)y=x*xout=y.mean()out.backward() #等效于torch.autograd.backward(out)print(x.grad)输出:tensor([[ 6.0000, 3.0000], [15.5000, 0.4500]原创 2021-08-02 20:04:44 · 89 阅读 · 0 评论