无法复制PyTorch的神经传递示例

时间:2018-03-24 18:56:31

标签: python deep-learning pytorch

我正在关注PyTorch's Neural Transfer的教程。但是,在下载Jupyter笔记本并运行它之后,我没有按预期获得输出。关于我做错了什么想法?

观察输出。

Observed Output

预期产出。

Expected Output

应该注意的是,我在以下环境中没有CUDA支持(在Mac笔记本电脑上)。

  • conda 4.5.0
  • python 2.7.17
  • jupyter 4.4.0
  • pytorch 0.3.1
  • torchvision 0.2.0

我注意到观察到的输出是128x128,预期的是512x512(尽管这种观察到的差异可能来自使用CPU或GPU的设置)。

观察到的样式转移的输出如下所示。

Building the style transfer model..
Optimizing..
run [50]:
Style Loss : 351316.375000 Content Loss: 53.188274

run [100]:
Style Loss : 318270.687500 Content Loss: 51.054543

run [150]:
Style Loss : 627803.375000 Content Loss: 59.788795

run [200]:
Style Loss : 351316.375000 Content Loss: 53.188274

run [250]:
Style Loss : 318270.687500 Content Loss: 51.054543

run [300]:
Style Loss : 627803.375000 Content Loss: 59.788795

预期输出如下所示。

Building the style transfer model..
Optimizing..
run [50]:
Style Loss : 0.147848 Content Loss: 0.470452

run [100]:
Style Loss : 0.043477 Content Loss: 0.343297

run [150]:
Style Loss : 0.035841 Content Loss: 0.315429

run [200]:
Style Loss : 0.031865 Content Loss: 0.306764

run [250]:
Style Loss : 0.030837 Content Loss: 0.302101

run [300]:
Style Loss : 0.030777 Content Loss: 0.299595

1 个答案:

答案 0 :(得分:0)

输出图像看起来就像我在保存之前没有钳制输出图像时的结果。