Pytorch
文章平均质量分 54
zhuiyuan2012
这个作者很懒,什么都没留下…
展开
-
model.show_result()导致jupyter lab提示服务似乎挂掉,但会立刻重启
实验1代码,不加model.show_result:%reload_ext autoreload%autoreload 2%matplotlib inlinefrom mmseg.apis import inference_segmentor, init_segmentorimport mmcvimport osos.environ[“KMP_DUPLICATE_LIB_OK”]=“TRUE”config_file = ‘mmsegmentation/configs/unet/fcn_une原创 2022-03-25 14:59:44 · 5026 阅读 · 0 评论 -
输出已训练好的模型特定层结构时只输出Sequential()
目的:加载已训练好的模型,输入图片,从前向过程中提取特定层输出结果。问题:通过模型的children()函数没法输出特定层结构实验过程:实验1:def predict(img_path):net = torch.load('./resnet18_.pth')#net = net.to(device)model2 = model()#print(model2)#print(net['net'])model2.load_state_dict(net["net"])print(model2原创 2022-03-22 22:45:58 · 625 阅读 · 0 评论 -
RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THCUNN/gen
pytorch报错:RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THCUNN/generic/ClassNLLCriterion.cu:224原因:模型输出的类别个数和读入的类别个数不一致导致的将输出类别个数修改为与输入类别个数一致即可。...原创 2021-10-18 11:23:45 · 431 阅读 · 0 评论 -
GPU很空闲总是提示GPU out of memeory
nvidia-smi 和watch -n 1 nvidia-smi命令显示GPU占用为0,没有进程使用GPU0上有11G空闲,但是训练模型时报错:提示:CUDA out of memory. Tried to allocate 1.12 GiB (GPU 0; 11.17 GiB total capacity; 9.99 GiB already allocated; 877.75 MiB free; 1.25 MiB cached)需要1.12G内存,但是不够。明明有11.17G啊!原创 2021-06-30 15:06:32 · 24700 阅读 · 9 评论 -
Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
报错:Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #2 'weight'这个是pytorch输出参数的类型问题:试过很多种方法,有种最有效的,修改如下:input=ipnut.type(torch.cuda.FloatTensor),转换下就好了...原创 2020-08-24 15:47:43 · 5003 阅读 · 0 评论 -
VAE学习材料
VAE论文和代码:https://github.com/AntixK/PyTorch-VAE原创 2020-08-13 16:53:11 · 209 阅读 · 0 评论 -
RuntimeError: CUDA error: device-side assert triggered
CUDA ERROR: device-side assert triggered at 问题及解决思路HuaYuuuu 2019-05-29 15:20:50 21384 已收藏 2版权cuda errorRuntimeError: cuda runtime error (59) : device-side assert triggered at ...我之前还以为是因为GPU抽风了引发的BUG,所以第一次没有在意,直接又重新开始运行了一次,但是第二次就发现程序在同样的地方断掉了,这..原创 2020-07-21 15:58:31 · 1605 阅读 · 1 评论 -
关于训练数据放batchsize个到内存
在模型训练过程中改变训练数据路径会报错:模型训练部分代码:#https://github.com/dhuQChen/DenseNetfrom model import *import torchvisionimport torchvision.transforms as transformsfrom torch.autograd import Variableimport matplotlib.pyplot as pltplt_loss = []def train(epoc.原创 2020-07-09 20:18:41 · 712 阅读 · 0 评论 -
pytorch查看模型的参数总量、占用显存量以及flops
from torchstat import statimport torchvision.models as modelsmodel = models.resnet152()stat(model, (3, 224, 224))原创 2020-07-09 16:02:28 · 11795 阅读 · 1 评论