pytorch编程
cg_nswdg
这个作者很懒,什么都没留下…
展开
-
【RuntimeError: set_sizes_and_strides is not allowed on a Tensor created from .data or .detach().】
模型时候,并没有这样的错误,而这个错误仅涉及到鉴别器的前向操作,并且,loss_gp的计算是包含了loss_D计算时所需的操作的,但loss_gp不会抛出这种错误(我排查过,十分确定)。尝试了requires_grad_(False)这种梯度截断和它推荐的解决办法,都没有解决问题,最后。大概意思是说用detach()时候不能使用某个功能。原创 2023-02-09 21:53:37 · 746 阅读 · 1 评论 -
远程命令行运行.py文件,python导入包提示No Module named XXX
在pycharm-run时候是正常的,但是服务器远程运行时候会提示no module named xxx。原创 2022-12-21 12:08:53 · 333 阅读 · 0 评论 -
pycharm安装opencv之后输入cv2不会自动联想函数
pycharm安装opencv之后输入cv2不会自动联想函数原创 2022-12-04 16:30:24 · 1902 阅读 · 3 评论 -
【mmseg 运行自定义数据集 提示‘MYDataset is not in the dataset registry‘】
mmseg error记录原创 2022-11-23 15:18:33 · 3076 阅读 · 2 评论 -
RuntimeError: one of the variables needed for gradient computation has been modified by an inplace
最后加上了detach,并且最先更新A网络的参数。虽然问题解决了,但不明白为什么当A放在最后更新时,就算先zero_grad,也会报第一个错误。原创 2022-11-08 22:36:24 · 1154 阅读 · 0 评论 -
【pytorch】CrossEntropyLoss Assertion `t >= 0 && t < n_classes` failed.
应用交叉熵计算多分类损失Assertion t >= 0 && t < n_classes failed.报错语句:loss_c = F.cross_entropy(c, cl.type(torch.cuda.LongTensor))/opt/conda/conda-bld/pytorch_1565272279342/work/aten/src/THCUNN/ClassNLLCriterion.cu:105: void cunn_ClassNLLCriterion_upd原创 2022-04-19 14:34:42 · 4286 阅读 · 7 评论 -
基于sklearn.metrics.roc_auc_score的二分类AUC计算方式
二分类,sklearn包roc_curve计算曲线下面积// 分割结果,单张图片AUC计算必须保证y_true(gt)是二值、整型的,y_test(out_net)是网络输出结果,float数据,表示网络将像素分为0/1类的对应概率。// code from sklearn.metrics import roc_curve, auc out_s = out_s.squeeze(0).detach().numpy() gt_s = gt_s.squeeze(0).cpu().numpy().as原创 2022-03-10 17:14:07 · 2894 阅读 · 0 评论 -
【Python】计算一组数据的信息熵-混乱度
python 计算信息熵网上看到用collections的Counter类计数的,但是Counter计数的dict_value不知道怎么转为数组,所以用numpy的bincount计数。bincount输入要求是整型,以array的格式返回数列的最大值最小值范围内每个数值出现的次数,没有出现的记0,返回array的长度是:max-min+1代码:def entropy(pixels): num = len(pixels) pixels = pixels.astype(int)原创 2021-12-01 10:21:08 · 2067 阅读 · 0 评论 -
【Pytorch测试时显卡占用率高,GPU利用率低,CUDA out of memory】
Pytorch测试时GPU利用率低,CUDA out of memorymodel很小,batchsize=1,watch -n -1 nvidia-smi;监视当前GPU使用情况,发现测试时GPU占用爆表,但是利用率只有1%,在测试时加入:def test: model.eval() with torch.no_grad(): test code... model.train()解决,可以正常运行了。...原创 2021-11-25 14:40:57 · 3104 阅读 · 0 评论