PacktPublishing / Deep-Learning-for-Computer-Vision

Deep Learning for Computer Vision, by Packt
MIT License
213 stars 167 forks source link

ValueError in Chapter08/1_style_transfer.py #8

Closed inyong37 closed 5 years ago

inyong37 commented 5 years ago

In Chapter 8, I modified vgg16_avg.py to match keras 2.0.

But I get an error on line 100 in 1_style_transfer.py, but I do not know how to fix it.

line 100 is as follows,

style_loss = sum(style_mse_loss(l1[0], l2[0]) for l1, l2 in zip(style_features, style_targets))

And the error is as follows,

ValueError: Dimensions must be equal, but are 64 and 128 for 'add_1' (op: 'Add') with input shapes: [64], [128].

I would appreciate it if you could tell me how to fix the error.