pytorch
文章平均质量分 57
DanielFaster
这个作者很懒,什么都没留下…
展开
-
pytorch 遇到OSError: [Errno 12] Cannot allocate memory错误的解决
在跑一段之前正常的代码,突然间就报[Errno 12] Cannot allocate memory ,通过排查内存方面的问题,最终解决原创 2022-04-07 13:55:47 · 13519 阅读 · 3 评论 -
ValueError: Not a TBLoader or TBPlugin subclass: <class ‘tensorboard_plugin_wit.wit_plugin_loader.Wh
解决使用tensorboardX过程中出现了的问题 ValueError: Not a TBLoader or TBPlugin subclass原创 2022-04-02 16:04:09 · 2046 阅读 · 0 评论 -
python log日志重复打印的问题(尝试全网无效,终解决)
实在找不出问题,就注释大法,排查下,这是最笨,但是在毫无思绪下,最有效的办法。原创 2022-04-01 10:53:23 · 2680 阅读 · 2 评论 -
Linux系统下pytorch官网conda命令安装pytorch的版本是cpu
用pytorch官网conda命令安装的pytorch的版本是cpu的解决方法原创 2022-03-31 17:03:22 · 3708 阅读 · 0 评论 -
Expected more than 1 value per channel when training, got input size torch.Size([1, **])
目录出现这种问题大概有两种情况如果上面步骤三还没解决问题原因分析先定位下问题出现这种问题大概有两种情况第一种情况是训练时候出现,另一种是测试模型的时候出现出现这种问题一般是你输入的数据只有一张图片例如 [1,3,256,256]步骤1 你试下改成 [2,3,256,256] 能否解决(如果无法加,请看下一个步骤),如果你是在训练模型,看步骤2 如果你是在测试模型,看步骤3步骤2 如果你是在训练模型,在获取数据集时,将DataLoader中drop_las...原创 2021-05-10 09:13:41 · 2969 阅读 · 0 评论 -
查询指定Python版本环境中包版本
多个版本的python,在使用的时候会遇到一些问题,例如安装,或者查询包版本信息等,下面我们将以问答的形式罗列一下回到问题情况问题1将包安装到指定Python版本环境中我的环境中有多个版本的python,有一个Python3.5,一个python3.6只是名字不同,现在我只想更新Python3.6中解决办法使用命令 python3 -m pip install torchvision==0.9.0问题2 查询指定环境下torchvision的版本解决办法使用命令 ...原创 2021-04-03 17:38:47 · 849 阅读 · 0 评论 -
cannot import name ‘__new_empty_tensor‘ 升级版本后依旧出现问题
问题描述原因说明这是因为包版本过低导致的,升级到torchvision 0.9.0就没有这个问题了后续问题我发现更新了版本依旧出现,后面发现因为我是多版本Python,运行的时候使用了默认环境,更新包在另一个环境,所有出现了问题解决办法原来使用 python3 xxx.py指定环境python3.6 xxxx.py...原创 2021-03-17 09:59:22 · 4477 阅读 · 0 评论 -
模型训练过程输出正常,跑验证集时某个分支输出效果很差(基于pytorch)
本文这是我宝贵的实战经验,看到的朋友请给我一打赞,非常感谢!!!问题说明最近我修改了一个模型,增加了一个分支。模型训练过程中各个分支输出正常,但是在跑验证集时某个分支输出效果很差,输出内容几乎是一副黑图,啥东西都没有。我做了网友给的两个解决办法,还是没能解决我的问题。一开始我很烦躁,因为之前在其他模型测试没有这个问题,代码也是差不多的,只是模型文件不同,所以没有什么思路。此时,我开启了网络寻答之路,网友们反馈的是1、可能是你数据没有归一化。但是我的数据已经归一化了。(最后发现是我的gt数原创 2020-10-09 20:03:52 · 1378 阅读 · 3 评论 -
加载模型出现 RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available()
今天在加载模型的时候出现了这个报错问题现象RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location='cpu' to map your storages to the CPU....原创 2020-08-11 20:52:34 · 9907 阅读 · 1 评论 -
加载模型出现 RuntimeError: Error(s) in loading state_dict for Model: Missing key(s) in state_dict
今天准备加载一个模型来测试的时候发现了一个问题,加载总是失败,报错是RuntimeError: Error(s) in loading state_dict for Model: Missing key(s) in state_dict"convd1.0.weight", "convd1.0.bias", "convd1.1.weight" 。咋一看,难道是因为我取值的问题,然后debug了一下,发现我的state_dict是符合要求的,但是为什么出现加载不了?问题代码 model = ..原创 2020-08-11 20:47:03 · 73583 阅读 · 93 评论 -
RuntimeError: CUDA out of memory 的解决
今晚在跑代码的时候出现这个问题,但是查看了一下GPU,命名内存是够的,后面发现虽然自己看到的内存是够的,但是真实的原因的确是因为内存不够导致,最后我调低了batchSize 问题就解决了!Traceback (most recent call last): File "train_SCRN.py", line 84, in <module> pred_sal, pred_edge = model(images) File "/home/ihavc/anaconda3...原创 2020-06-02 00:34:25 · 9297 阅读 · 0 评论 -
F.pad 的理解
本文转自http://www.luyixian.cn/news_show_324628.aspx这位作者写得非常棒!!!1. F.pad函数定义F.pad是pytorch内置的tensor扩充函数,便于对数据集图像或中间层特征进行维度扩充,下面是pytorch官方给出的函数定义。torch.nn.functional.pad(input, pad, mode='constant', value=0)函数变量说明:input 需要扩充的tensor,可以是图像数据,抑或是特征..转载 2020-05-26 16:40:03 · 10779 阅读 · 12 评论