Pytorch实现风格迁移代码提示错误

2021-01-07 11:14发布

站内问答 / Python
3937 7 12

Traceback (most recent call last):  File test.py, line 84, in     main()  File test.py, line 64, in ...

Traceback (most recent call last):
  File "test.py", line 84, in 
    main()
  File "test.py", line 64, in main
    res = torch.cat([c_denorm, out_denorm], dim=0)
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 459 and 456 in dimension 2 at C:\w\1\s\windows\pytorch\aten\src\TH/generic/THTensor.cpp:689

在测试新图片的时候，总是提示这种错误，测试图片只有RGB三个通道，请教大神解答

发送
7条回答
思禹小姐姐y
2楼 · 2021-01-07 15:34

VGG16下载链接：https://download.pytorch.org/models/vgg16-397923af.pth

然后放入对应的路径就可以了。

或者 我们设置将所有含有nn.InstanceNorm2d都设置track_running_stats=True

Kindery
3楼 · 2021-01-07 16:21

Pytorch实现风格迁移代码提示错误

按照命令行提示，我们设置将所有含有nn.InstanceNorm2d都设置track_running_stats=True


是开心果呀 - 热爱生活
4楼 · 2021-01-07 16:53

报错中说InstanceNorm2d有问题，让我们设置track_running_stats=True。

我们看一下对应报错行的代码，先看一部分含有InstanceNorm2d的代码，如下

  1. self.initial_layers = nn.Sequential(

  2. ConvLayer(3, 32, kernel_size=9, stride=1),

  3. nn.InstanceNorm2d(32, affine=True),

  4. nn.ReLU(True),

  5. ConvLayer(32, 64, kernel_size=3, stride=2),

  6. nn.InstanceNorm2d(64, affine=True),

  7. nn.ReLU(True),

  8. ConvLayer(64, 128, kernel_size=3, stride=2),

  9. nn.InstanceNorm2d(128, affine=True),

  10. nn.ReLU(True),

  11. )

按照命令行提示，我们设置将所有含有nn.InstanceNorm2d都设置track_running_stats=True

不吃鱼的猫
5楼 · 2021-01-08 09:52

设置将所有含有nn.InstanceNorm2d都设置track_running_stats=True

茄子酱
6楼 · 2021-01-08 10:04

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 stats by default since 0.4.0. Please remove these keys from state_dict. If the running stats are actually needed, instead set track_running_stats=True in InstanceNorm2d to enable them. See the documentation of InstanceNorm2d for details.

报错中说InstanceNorm2d有问题，让我们设置track_running_stats=True。

我们看一下对应报错行的代码，先看一部分含有InstanceNorm2d的代码，如下。

self.initial_layers = nn.Sequential(ConvLayer(3, 32, kernel_size=9, stride=1),nn.InstanceNorm2d(32, affine=True),nn.ReLU(True),ConvLayer(32, 64, kernel_size=3, stride=2),nn.InstanceNorm2d(64, affine=True),nn.ReLU(True),ConvLayer(64, 128, kernel_size=3, stride=2),nn.InstanceNorm2d(128, affine=True),nn.ReLU(True),)

按照命令行提示，我们设置将所有含有nn.InstanceNorm2d都设置track_running_stats=True，如下。

self.initial_layers = nn.Sequential(ConvLayer(3, 32, kernel_size=9, stride=1),nn.InstanceNorm2d(32, affine=True, track_running_stats=True),nn.ReLU(True),ConvLayer(32, 64, kernel_size=3, stride=2),nn.InstanceNorm2d(64, affine=True, track_running_stats=True),nn.ReLU(True),ConvLayer(64, 128, kernel_size=3, stride=2),nn.InstanceNorm2d(128, affine=True, track_running_stats=True),nn.ReLU(True),)

 


三岁奶猫
7楼 · 2021-01-08 11:03

Pytorch实现风格迁移代码提示错误

按照命令行提示，我们设置将所有含有nn.InstanceNorm2d都设置track_running_stats=True


lucky璐呀
8楼 · 2021-12-01 10:24

Pytorch实现风格迁移代码提示错误；按照命令行提示，我们设置将所有含有nn.InstanceNorm2d都设置track_running_stats=True


相关问题推荐

  • python中\n 是什么意思?2020-04-16 09:41
    回答 3

    换行。比如，print hello\nworld效果就是helloworld\n就是一个换行符。\是转义的意思，'\n'是换行，'\t'是tab，'\\'是，\ 是在编写程序中句子太长百，人为换行后加上\但print出来是一整行。...

  • python中有哪些简单的算法2020-06-10 17:06
    回答 42

    十种常见排序算法一般分为以下几种：（1）非线性时间比较类排序：a. 交换类排序（快速排序、冒泡排序）b. 插入类排序（简单插入排序、希尔排序）c. 选择类排序（简单选择排序、堆排序）d. 归并排序（二路归并排序、多路归并排序）（2）线性时间非比较类排序：...

  • 人工智能好找工作吗？2020-10-11 17:45
    回答 70
    已采纳

    前景很好，中国正在产业升级，工业机器人和人工智能方面都会是强烈的热点，而且正好是在3~5年以后的时间。难度，肯定高，要求你有创新的思维能力，高数中的微积分、数列等等必须得非常好，软件编程(基础的应用最广泛的语言：C/C++)必须得很好，微电子(数字电...

  • 迭代器和生成器的区别?2020-06-10 09:20
    回答 28

    迭代器与生成器的区别：（1）生成器：生成器本质上就是一个函数，它记住了上一次返回时在函数体中的位置。对生成器函数的第二次（或第n次）调用，跳转到函数上一次挂起的位置。而且记录了程序执行的上下文。生成器不仅记住了它的数据状态，生成器还记住了程序...

  • python中title( )是什么意思python 2020-09-02 11:59
    回答 9

    python中title( )属于python中字符串函数，返回’标题化‘的字符串，就是单词的开头为大写，其余为小写

  • python中cnt是什么意思python 2020-09-02 11:59
    回答 6

    第一种解释：代码中的cnt是count的简称，一种电脑计算机内部的数学函数的名字，在Excel办公软件中计算参数列表中的数字项的个数；在数据库( sq| server或者access )中可以用来统计符合条件的数据条数。函数COUNT在计数时，将把数值型的数字计算进去；但是...

  • python中的data[:, :-1]和data2020-12-30 11:21
    回答 1

    head是方法，所以需要取小括号，即dataset.head()显示的则是前5行。data[:, :-1]和data[:, -1]。另外，如果想通过位置取数据，请使用iloc，即dataset.iloc[:, :-1]和dataset.iloc[:, -1]，前者表示的是取所有行，但不包括最后一列的数据，结果是个DataFrame。...

  • Python入门简单吗2021-09-23 13:21
    回答 45

    挺简单的，其实课程内容没有我们想象的那么难、像我之前同学，完全零基础，培训了半年，直接出来就工作了，人家还在北京大公司上班，一个月15k，实力老厉害了

  • python的内建函数在什么位置内建函数 2022-04-02 16:28
    回答 4

    Python针对众多的类型，提供了众多的内建函数来处理(内建是相对于导入import来说的，后面学习到包package时，将会介绍)，这些内建函数功用在于其往往可对多种类型对象进行类似的操作，即多种类型对象的共有的操作；如果某种操作只对特殊的某一类对象可行，Pyt...

  • Python中注释前面加上引号是什么意思？python注释 2022-03-11 18:21
    回答 8

     相当于 ... 这里不是注释

  • 除了#TODO在pycharm还有哪些高亮的注释？python注释 2022-03-11 18:20
    回答 4

    还有FIXME

  • python写入数据到excel怎么保留excelExcel功能 2022-02-09 18:56
    回答 3

    python的两个库:xlrd和xlutils。 xlrd打开excel,但是打开的excel并不能直接写入数据,需要用xlutils主要是复制一份出来,实现后续的写入功能。

  • Python中注释的分类python注释 2022-01-17 18:20
    回答 8

    单行注释：Python中的单行注释一般是以#开头的，#右边的文字都会被当做解释说明的内容，不会被当做执行的程序。为了保证代码的可读性，一般会在#后面加一两个空格然后在编写解释内容。示例：#  单行注释print(hello world)注释可以放在代码上面也可以放在代...

  • 如何才能实现python文件读取中对注释行的识别python注释 2022-01-06 15:06
    回答 2

    主要是按行读取，然后就是写出判断逻辑来勘测行是否为注视行，空行，编码行其他的：import linecachefile=open('3_2.txt','r')linecount=len(file.readlines())linecache.getline('3_2.txt',linecount)这样做的过程中发现一个问题，...

  • pycharm 为什么按ctrl/不会取消注释而是python注释 2022-01-06 15:04
    回答 4

    或许是里面有没被注释的代码

  • 零基础学习Python需要多长时间，有没有推荐的课2021-10-20 17:23
    回答 26

    自学的话要看个人情况，可以先在B站找一下视频看一下

没有解决我的问题，去提问