search

fzliu / style-transfer

An implementation of "A Neural Algorithm of Artistic Style" by L. Gatys, A. Ecker, and M. Bethge. http://arxiv.org/abs/1508.06576.
1.54k stars 387 forks source link

batch_size #30

Open pp072 opened 8 years ago

pp072 commented 8 years ago

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

fzliu commented 8 years ago

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 commented 8 years ago

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 commented 8 years ago

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