ProGamerGov / neural-style-pt

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

Image Size to VRAM Relationship #54

Open spot92 opened 4 years ago

spot92 commented 4 years ago

How large of an image can people make using the L-BFGS optimizer? And how much VRAM are you working with? I max out somewhere around 1664 when using -cudnn_autotune on a 1080ti with 11 gigs.

spot92 commented 4 years ago

I have a 1080 ti and can get to 1500 using the L-BFGS, can then upres to 2560 using ADAM. This does vary depending on overall dimensions.