vgg model does not calculate the style and content loss

ProGamerGov / neural-style-pt

PyTorch implementation of neural style transfer algorithm

MIT License

838 stars 175 forks source link

vgg model does not calculate the style and content loss #4

Closed eric11eca closed 5 years ago

eric11eca commented 5 years ago

When I use the vgg model, it does not calculate the content and style loss. It only prints out the total loss. The NIN model works just fine.

ProGamerGov commented 5 years ago

Can you please elaborate on this issues your having? What PyTorch version you are using? What are the outputs for the content/style loss? Is the output still stylized?

ProGamerGov commented 5 years ago

If this issue still exists, then please reopen the issue.