neural-style icon indicating copy to clipboard operation
neural-style copied to clipboard

Reducing the usage of VRAM means damniting the quality of image?

Open linrio opened this issue 7 years ago • 2 comments

I tried to reduce the usage of VRAM (GTX 1070, 8GB). I replace the VGG-16(19) model with NIN model, and I replace the L-BFGS with adam. Reference from memory usage. However, the quality of image is more inferior than that with default method (VGG-16, L-BFGS). See #328 . How to ensure the quality of image meanwhile reduce the usage of VRAM?

linrio avatar Apr 18 '17 04:04 linrio

Have you look at this wiki post yet?

vkbs avatar Jul 11 '17 05:07 vkbs

@linrio Adam appears to use bad parameters by default in Neural-Style. If you modify the Adam specific parameters, you can make Adam a lot more compatible with L-BFGS.

You can find more information here: https://github.com/jcjohnson/neural-style/wiki/Fine-Tuning-The-Adam-Optimizer

ProGamerGov avatar Feb 23 '18 00:02 ProGamerGov