pytorch
ZHAGNQ
这个作者很懒,什么都没留下…
展开
-
RuntimeError: one of the variables needed for gradient computation has been modified by an inplace
找到网络模型中的inplace操作。x = x + 1 # not inplacex += 1 # inplace把第二种写法,替换成第一种。但是替换完你会发现,还是报这个错误,因为,relu中也有inplace操作,这时,你需要将relu中的inplace赋为False,for example,self.embedding_net2[0].inplace = False...原创 2019-07-09 23:11:29 · 628 阅读 · 0 评论 -
读SPED_900数据集下各文件夹图片并显示
读SPED_900数据集下各文件夹图片并显示,以选取用于测试的图片import osimport pdbimport matplotlib.pyplot as pltimport matplotlib.image as mpimg# from PIL import Imagepath = r'D:\SPED_900'files = os.listdir(path)plt....原创 2019-09-05 00:16:43 · 460 阅读 · 2 评论 -
可视化网络中间图特征文件
在网络的完成后,由于画图需要,我们可能需要可视化中间特征图,可使用以下代码import osfrom torch.utils.data import Dataset, DataLoaderfrom PIL import Imagefrom torchvision import transformspath = 'path'def default_loader(img_pat...原创 2019-09-05 00:27:02 · 701 阅读 · 0 评论