leongatys / PytorchNeuralStyleTransfer

Implementation of Neural Style Transfer in Pytorch
MIT License
421 stars 103 forks source link

How the "vgg_conv.pth" is generated? #1

Closed panovr closed 7 years ago

panovr commented 7 years ago

I found that "vgg_conv.pth" only has the conv layers of the original vgg net, and the fc layers are removed. May I ask that how this model is generated?

leongatys commented 7 years ago

Hope that helps, Leon

panovr commented 7 years ago

Got it, thanks!

michaelhuang74 commented 6 years ago

@leongatys Thanks for the steps. Could you kindly share the code that converts the caffe model to pytorch model?