【PyTorch】用PyTorch实现风格迁移所遇到的问题解决方案

最新推荐文章于 2022-04-15 14:03:20 发布
最新推荐文章于 2022-04-15 14:03:20 发布
阅读量1.6k 收藏 1
点赞数 3
分类专栏： PyTorch 文章标签： PyTorch 风格迁移
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/giftedpanda/article/details/104183859
版权
本文介绍了在使用PyTorch实现风格迁移过程中遇到的两个常见错误：1. 存储大小错误，原因是缺少预训练模型VGG16，解决方案是下载并放置到正确路径；2. InstanceNorm2d运行状态缓冲区错误，解决方法是更新InstanceNorm2d模块的track_running_stats参数为True。通过这两个步骤，可以成功解决运行时的问题。
摘要由CSDN通过智能技术生成

1：RuntimeError: storage has wrong size: expected -1122305894 got 128

我们来看一下对应行的代码：

style_model.load_state_dict(t.load(opt.model_path, map_location=lambda _s, _: _s))

很明显这是加载预训练好的模型出错了，笔者再测试了一下需要用到的模型，结果发现没有VGG16，于是下载了VGG16放入对应路径下就解决了这个问题。

VGG16下载链接：https://download.pytorch.org/models/vgg16-397923af.pth
2：Unexpected running stats buffer(s) "upsample_layers.4.running_mean" and "upsample_layers.4.running_var" for InstanceNorm2d with track_running_stats=False. If state_dict is a checkpoint saved before 0.4.0, this may be expected because InstanceNorm2d does not track running stat

最低0.47元/天 解锁文章
我是一只计算鸡
关注
专栏目录
利用Pytorch进行迁移学习时常遇到的问题及解决方案
AI天才研究院
08-09 810
作者：禅与计算机程序设计艺术 1.简介 在机器学习的过程中，数据集往往都非常大，如果直接使用大规模的数据集训练模型，将耗费大量的时间。因此，如何有效地利用小样本数据集进行迁移学习（transfer learning）成为了一个重要问题。在PyTorch中，可以使用迁移学习的库例如torchvision，
PyTorch深度学习实战（38）——StyleGAN详解与实现
最新发布
盼小辉丶的博客
03-05 3334
StyleGAN 是生成对抗网络 的变体，是一种无监督学习模型，用于生成逼真且高分辨率的图像。StyleGAN 能够生成非常高分辨率人脸图像的关键在于，在增加分辨率的步骤中逐步增加生成网络和判别网络的复杂性，以便在每一步中，两个模型都可以很好地完成任务。本节中，介绍了如何通过确保每个分辨率的特征由一个独立的输入(称为风格向量)来控制，从而操纵生成图像的风格，以及如何通过将图像之间的风格进行交换来操作不同图像的风格。
RuntimeError: storage has wrong size: expected XXX got XXX 以及 多卡GPU训练转到多卡GPU测试
qq_42035274的博客
04-15 3588
在训练时，我的代码出现了以下问题： Traceback (most recent call last): File "/root/PycharmProjects/test.py", line 8, in <module> model_dict = torch.load("0.pth") File "torch/serialization.py", line 529, in load return _legacy_load(opened_file, map_locati.
风格迁移 RuntimeError: storage has wrong size: expected 16825408 got 256
longshaonihaoa的博客
08-15 1万+
闲着试一下风格迁移，pytorch0.4，python3，windows系统 尝试使用了： http://www.pytorchtutorial.com/pytorch-style-transfer/ https://blog.csdn.net/zzlyw/article/details/78849250 提供的代码，在此先感谢两篇文章的作者 但是都遇到了 ...(之前省略) ...
pytorch官网预训练模型百度云下载 VGG16,Densnet169，inception_v3
工科π的博客
06-29 8159
在深度学习领域采用预训练的模型参数进行迁移学习往往会得到事半功倍的效果。但是在使用pytorch加载预训练模型是往往会在下载模型时报错。VGG16报错：https://download.pytorch.org/models/vgg16-397923af.pth 模型保存路径：/home/主机名/.cache/torch/checkpoints/vgg16-397923af.pth， inception_v3报错：https://download.pytorch.org/models/inception_
RuntimeError: storage has wrong size: expected -1112053942 got 589824
xiaxiahhu111的博客
03-14 1533
RuntimeError: storage has wrong size: expected -1112053942 got 589824 在自己电脑测试模型时，遇到了这个问题。 解决方案：可能是之前.pth文件没有下载好，重新下载再运行就好了。 参考链接
解决pytorch预训练模型下载速度慢的问题
zhyue77yuyi的博客
07-18 7359
在远程跑代码的时候，我需要 from torchvision.models.resnet import resnet50 然而远程服务器上下载太慢太慢了，有时还会下载进度突然不动了，这……心累啊…… 解决方法： 1. 去以下网址下载你需要的模型： 1 Resnet: 2 3 model_urls = { 4 'resnet18': 'https://download.pytorch.org/models/resnet18-5c106cde.pth', 5 'resn
图片风格迁移系统.zip
10-14
4. README说明：提供了项目的概述、安装依赖、运行指南以及可能遇到的问题解决方案，对于初学者非常有帮助。 在PyTorch中，可以使用预训练的VGG网络（如VGG19）来提取风格和内容特征。VGG网络在ImageNet数据集上...
windows环境离线配置安装pytorch+遇到的坑
12-20
遇到问题时，查阅官方文档、社区论坛或者搜索引擎是寻找解决方案的有效途径。 总之，离线安装PyTorch需要提前准备，明确系统配置，选择合适的版本，并利用有效的网络资源。遵循正确的步骤和注意事项，通常可以避免...
lambda使用详解
09-03
java8中关于lambda和stream流的重点技术，可以全面的学习lambda表达式常用的知识点及stream流的代码开发的高效性。
pytorch踩坑记
qq_43994782的博客
08-11 2431
1、关于cpu cuda Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same 英语4级425分的我一看就懂了，数据和模型没在一个地儿呗， 模型在cuda上，但是数据在cpu上，所以要移动一下 data.to(torch.device('cuda')) 然后。。。。 Input type (torch.FloatTensor) and weight type (torch.cud
RuntimeError: storage has wrong size | Unable to load weights from pytorch checkpoint file
HeavenWalker
11-05 2183
报错1: RuntimeError: storage has wrong size: expected 768 got 2359296 报错2: Unable to load weights from pytorch checkpoint file 报错场景: 使用torch.nn.parallel.DistributedDataParallel进行单机多卡模型训练, 保存的bert模型在读取的时候报上述错误, 原因是保存模型的时候使用多线程保存, 猜想是多卡同时写入, 文件在保存过程中出了问题 解决
torch.load()加载模型时报错
weixin_43820848的博客
09-10 1万+
torch.load()加载模型时出现如下错误 Traceback (most recent call last): File "demo_syncnet.py", line 26, in <module> s.loadParameters(opt.initial_model); File "/media/cj/75bb371d-0b6d-4995-bb72-060a4...
Pytorch预训练模型下载并加载（以VGG为例）自定义路径
热门推荐
肥宅Sean
06-08 1万+
简述 一般来讲，Pytorch用torchvision调用vgg之类的模型话，如果电脑在cache（Pytorch硬编码的一个地址）（如果在环境变量中添加了TORCH_HOME 和TORCH_MODEL_ZOO的话，就是在这两个位置的联合的路径下，比如TORCH_MODEL_ZOO\model）否则就是在TORCH_HOME\models或者是~/.torch/models 比如，我的就是C:\Users\lijy2/.torch\models\vgg11-bbd30ac9.pth。 这很有可能并不是我们想
Pytorch深度学习常用预训练网络模型的下载地址
qq_27245709的博客
07-22 4409
转载自https://blog.csdn.net/sumaliqinghua/article/details/90903590 Resnet: model_urls = { 'resnet18': 'https://download.pytorch.org/models/resnet18-5c106cde.pth', 'resnet34': 'https://download...
Pytorch迁移学习小技巧 以及 Pytorch小技巧的一些总结
fmbao的博客
05-02 1万+
迁移学习技巧 内容概要： 迁移学习的概念 Pytorch预训练模型以及修改 不同修改预训练模型方式的情况 一些例子：只针对dense layer的重新训练 ，冻结初始层的权重重新训练 迁移学习的概念 神经网络需要用数据来训练，它从数据中获得信息，进而把它们转换成相应的权重。这些权重能够被提取出来，迁移到其他的神经网络中，我们“迁移”了这些学来的特征，就不需要从零开始训练一个神经...
深度学习基础知识（三）--- Normalization 规范化 操作
朝花&夕拾
12-29 5350
本文部分过程和图片来源于以下参考资料 参考资料： 知乎专栏 - 张俊林 - 深度学习中的Normalization模型 - https://zhuanlan.zhihu.com/p/43200897 知乎专栏 - Juliuszh - 详解深度学习中的Normalization，BN/LN/WN - https://zhuanlan.zhihu.com/p/33173246 CSDN ...
Pytorch InstanceNorm2d Feature数与前一层输入不匹配但不报错
NLOS的博客
07-08 4780
问题描述 有一个能够正常运行的网络，其中有几层是这样的： XXXNet( (Encoder): Sequential( ... ... (9): Conv2d(256, 512, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1), bias=False) (10): InstanceNorm2d(1024, eps=1e-05, momentum=0.1, affine=False, track_running_st
Pytorch+CNN+迁移学习金属缺陷检测项目
资源摘要信息: "本项目是一个基于Pytorch框架，使用卷积神经网络（CNN）和迁移学习技术来实现金属表面缺陷检测的Python源码。源码包含完整的功能，并经过了稳定可靠性的验证。项目面向的用户群体广泛，包括但不限于...
评论 4
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值