Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

vgg without gpu #55

Open
maciejjaskowski opened this issue Apr 5, 2016 · 6 comments
Open

vgg without gpu #55

maciejjaskowski opened this issue Apr 5, 2016 · 6 comments

Comments

@maciejjaskowski
Copy link

All networks in the zoo use DNN. Is there a way to use a pretrained model if I don't have GPU installed ?

@f0k
Copy link
Member

f0k commented Apr 5, 2016

Yes, change Conv2DDNNLayer to Conv2DLayer and make sure you pass flip_filters=False in the convolution layer constructors. Then change MaxPool2DDNNLayer to MaxPool2DLayer.

@maciejjaskowski
Copy link
Author

Thanks!

@maciejjaskowski
Copy link
Author

Wait a second @f0k. Argument flip_filters does not exist in Conv2DLayer but the vgg16.py sets flip_filters=False. How should I understand your suggestion?

@f0k
Copy link
Member

f0k commented Apr 6, 2016

You will need to update to the bleeding-edge version of Lasagne for this: http://lasagne.readthedocs.org/en/latest/user/installation.html#bleeding-edge-version

@maciejjaskowski
Copy link
Author

Ok, that worked, thanks again. Do you think a 'cpu' argument could be included in lasagne recipes ?

@f0k
Copy link
Member

f0k commented Apr 8, 2016

We could simply change the file as I suggested to you, it will still use cuDNN when it's available. However, examples would stop working for Lasagne 0.1, so we should wait until we managed to release 0.2.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants