search

ProGamerGov / neural-style-pt

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

Support for additional pre trained models #93

Open Vigilence opened 1 year ago

Vigilence commented 1 year ago

Hello!

I would like to experiment with different models. Can you please share what steps I would need to take to add support for other models such as these?

ProGamerGov commented 1 year ago

@Vigilence The models you suggest are using ResNet architecture, which can require extra work to use for things like style transfer. neural-style-pt uses VGG models by default, but you can add support for them in the modelSelector function in Caffeloader.