search

ProGamerGov / neural-style-pt

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

Updated CaffeLoader.py #6

Closed rrmina closed 5 years ago

rrmina commented 5 years ago

Fixed error in #5

This is just a quick workaround, since style transfer doesn't use the classifier network anyway.

ProGamerGov commented 5 years ago

@iamRusty Did you download the model directly from: https://github.com/jcjohnson/pytorch-vgg, or did you use the included download script here: https://github.com/ProGamerGov/neural-style-pt/blob/master/models/download_models.py?