Pytorch
前进ing_嘟嘟
记录那些年我们淌过的坑,愿能撑人一把,也有人撑自己一把
展开
-
经过test之后再trainRuntimeError: element 0 of tensors does not require grad and does not have a grad_fn
我的程序训练一段时间,进行test输出,继续训练的时候出现标题所述问题。我遇到的问题不能通过loss.requires_grad_(True)解决。后发现原因是,我传入test的网络模型是用于train的网络模型实例。该问题的解决方案是不要将用于train的实例传入test,即假设net = Model.net1(),则不要test不要传入net,而要传入Model.net1()...原创 2022-03-07 15:35:06 · 364 阅读 · 1 评论 -
从txt或者log文件中提取特定字符所在行,可用于loss曲线绘制
import matplotlib.pyplot as pltimport numpy as npline = []with open(r"./loss.log", encoding='utf-8') as f: # 从log文件中读出数据 for line1 in f: line.append(line1) datalist = []for item in line: strl = item.split() datalist.append(str.原创 2020-08-26 11:51:00 · 669 阅读 · 0 评论 -
python 找到文件夹下所有的子文件夹下某类文件,并记录路径
file = './myfile/'for f inglob.glob(os.path.join(file, "*/*.jpg")): # 根据文件类型替换jpg,如文件深度更深,通过添加*/解决 print(f)原创 2020-08-26 11:04:21 · 764 阅读 · 0 评论 -
argument 'indices' must be tuple of Tensors, not Tensor
当我做loss backward的时候,遇到这个问题。先是查到,这个问题可以通过autograd.variable解决,我将最后得到loss的中间变量加上之后觉得无法利用之前的信息手写双线性插值也并不能解决这个问题在pytorch社区发现类似问题,有人提出,不用index,而是用 gather, index_put_ , masked_scatter_ or grid_sample ,给...原创 2019-05-05 21:29:28 · 2562 阅读 · 0 评论 -
Pytorch Tensor 输出为txt和mat格式
假设result1为tensor格式,首先将其化为array格式(注意只变成numpy还不行),之后存为txt和mat格式import scipy.io as ioresult1 = np.array(result1)np.savetxt('npresult1.txt',result1)io.savemat('save.mat',{'result1':result1})...原创 2019-04-16 15:20:51 · 27824 阅读 · 1 评论 -
Pytorch 函数查看
刚接触pytorch,对其结构之类的都不熟悉。在改一个程序的时候,需要将RGB图改成灰度图,所以在torchvision.transforms.Compose中加入 transforms.ToPILImage(), transforms.Grayscale(num_output_channels=3), transforms.ToTensor(),...原创 2019-04-09 17:05:59 · 4362 阅读 · 1 评论 -
Pytorch 查看数据类型
如果数据为tensor,用type(object)查看类型,用object.dtype查看数据的具体类型如果数据为numpy,PIL image等,也可以用type(object)来查看数据类型原创 2019-04-08 16:54:14 · 7479 阅读 · 0 评论 -
Pytorch torchvision.transforms.Compose 易错点
先看如下代码:import osimport torchimport torch.nn.functional as Fimport torch.nn.parallelimport torch.backends.cudnn as cudnnimport torch.optimimport torchvisionimport torch.utils.dataimport torc...原创 2019-04-08 16:41:08 · 5643 阅读 · 0 评论 -
Pytorch中常见函数
Pytorch中常见函数nn.Linear(input_feature, output_feature)他的输入和输出需要满足的是:输入input(A, ..., input_feature),输出output(B, ..., output_feature)nn.Conv2D(in_channels, out_channels, kernel_size, stride=1, paddin...原创 2019-04-10 16:20:27 · 1086 阅读 · 0 评论