PyTorch
「已注销」
这个作者很懒,什么都没留下…
展开
-
pytorch 保存模型,加载预训练模型问题
在写pytorch代码时,遇到问题:加载预训练模型时在验证集上测试的psnr结果与训练时验证集的psnr差异特别大。源代码:pretrained_dict = torch.load('epochG_515.pth')net.load_state_dict(pretrained_dict)net = prepare(net)valdata = Data(root=os.path.join(args.dir_data, args.data_val), args=args, train=False原创 2020-08-04 20:04:45 · 708 阅读 · 0 评论 -
输出list的大小
import numpy as npseq = []for i in range(10): seq.append(i)print(np.array(seq).shape)参考链接原创 2019-11-12 09:38:33 · 2477 阅读 · 0 评论 -
ERROR: Unexpected bus error encountered in worker. This might be caused by insufficient shared memor
运行EDSR代码时,出现如下错误:解决方案: dataloader.DataLoader( testset, batch_size=1, shuffle=False, pin_memory=not a...原创 2019-11-06 10:40:01 · 2266 阅读 · 1 评论 -
深度学习框架PyTorch:入门与实践 学习(一)
从今天起要从头开始学习PyTorch了,在此记下笔记。PyTorch 入门第一步Tensorimport Torch as tx = t.tensor(5,3)#生成5*3的矩阵print(x.size())# 输出x的维度print(x.size()[0]) #输出x的第0维print(x.size(0))# 输出x的第0维y = t.rand(5,3)# 生成0-1之...原创 2019-09-20 19:47:43 · 949 阅读 · 0 评论 -
深度学习框架PyTorch:入门与实践 学习(二)
Tensor和Autograd------TensorTensor创建:Tensor(*size),创建时不会立马分配空间,使用时才会分配,而其他均创建时立马就分配。 ones(*sizes) zeros(*sizes) eye(*sizes)对角线为1其他为0 arange(s,e,step)从s到e步长为step linspace(s,e,steps)从s到e,均匀切分成...原创 2019-09-23 21:29:47 · 269 阅读 · 0 评论 -
深度学习框架PyTorch:入门与实践 学习(四)
PyTorch中常用工具torchvisionmodels:提供深度学习中各种经典的网络结构以及预训练好的模型,包括AlexNet, VGG, ResNet, Inception datsets:提供常用的数据集加载,设计上均继承torch.utils.data.dataset,主要包括MNIST,CIFAR10/100,ImageNet,COCO transforms:提供常用的数据...原创 2019-09-24 11:09:38 · 282 阅读 · 0 评论 -
深度学习框架PyTorch:入门与实践 学习(三)
nn.Module用nn.Module实现全连接层import torch as tfrom torch import nnfrom torch.autograd import Variable as Vclass Linear(nn.Module): def __init__(self, in_features, out_features): super...原创 2019-09-28 22:07:34 · 281 阅读 · 0 评论