![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
qq_30852205
这个作者很懒,什么都没留下…
展开
-
pytorch测试过程中out of memory解决方法
在已经加入model.eval()情况下仍然没有解决,基本代码如下,因为需要计算test数据整体的评价结果,所以需要将probs存在list中,这会导致显存持续增加,故将probs等变量detach,如此测试时显存不再递增model.eval()probs, labels, losses = [], [], []for batch in self.batch_generator(): x,...原创 2020-03-01 17:52:22 · 1627 阅读 · 0 评论 -
cusparse.h: No such file or directory compilation terminated.
在使用python setup.py build develop安装awesome-semantic-segmentation-pytorch碰到了如下问题:In file included from /xxx/awesome-semantic-segmentation-pytorch/core/nn/csrc/cuda/syncbn_cuda.cu:5:0:/usr/local/lib/py...原创 2019-08-02 14:16:00 · 6414 阅读 · 2 评论 -
Pytoch CrossEntropyLoss类
[Pytorch] CrossEntropyLoss类官方注释这是针对多分类问题的损失函数注意:input不需要做normalized,直接輸入原始值就行(不需要提前softmax)损失函数:(其中包含了softmax的步骤)loss(x,class)=−log(exp(x[class])∑jexp(x[j]))=−x[class]+log(∑jexp(x[j]))\tex...原创 2019-08-06 19:22:36 · 336 阅读 · 0 评论 -
DataLoader worker (pid 173) is killed by signal: Bus error(Docker中用Pytorch多workers读取Data Loader)
DataLoader worker (pid XXXX) is killed by signal: Bus error(Pytorch多workers读取Data Loader)环境Docker Container中Pytorch 1.1Python 3.5问题背景在Docker的container中使用Pytorch 的DataLoader时,设置了num_worker>0...原创 2019-08-07 09:56:45 · 2762 阅读 · 0 评论 -
Pytorch Softmax
Pytorch的torch.nn.functional中有一个softmax方法,通常传入一个tensor和dim(即softmax所作用的维度,默认为0),如一个简单的例子:import torchimport torch.nn.functional as F# Example1t = torch.Tensor([1, 2, 3])F.softmax(t)# Out: tensor...原创 2019-08-28 15:58:44 · 2884 阅读 · 0 评论