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

batch_size #30

Open
pp072 opened this issue May 2, 2016 · 3 comments
Open

batch_size #30

pp072 opened this issue May 2, 2016 · 3 comments

Comments

@pp072
Copy link

pp072 commented May 2, 2016

Hi! Where I can change batch_size? I can't compute on gtx 960 4GB with vgg16 model ( !

@fzliu
Copy link
Owner

fzliu commented May 2, 2016

If memory is becoming a limitation for you, try reducing the image size (a smaller model like GoogleNet would also work).

With that being said, I'm surprised that 4GB wasn't enough for VGG16. What options are you using to generate the final image?

@pp072
Copy link
Author

pp072 commented May 3, 2016

Now I use resolution 512 and 1024 interations and it's work. But I want to use resolution 1024X1024. (style.py -s s7.jpg -c 3.jpg - m vgg16 -g 0 -l 512)

@fzliu
Copy link
Owner

fzliu commented May 4, 2016

If you want a higher-res image, try using a smaller model, or run it on the CPU.

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