search

ProGamerGov / neural-style-pt

PyTorch implementation of neural style transfer algorithm
MIT License
833 stars 178 forks source link

why use caffe vgg model #68

Closed happyhugo closed 3 years ago

happyhugo commented 4 years ago

why not use pytorch give the default vgg model?

ProGamerGov commented 4 years ago

@happyhugo When I was testing the different models, I found that the Caffe models would perform better the Torchvision models. Other individuals like Leon Gatys also appear to have come to the same conclusion.