search

ProGamerGov / neural-style-pt

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

Calculator for maximum image size #69

Open ArcticWinterSturm opened 4 years ago

ArcticWinterSturm commented 4 years ago

Is there a potential to make a tool to calculate maximum image size that can be archieved given certain GPU memory? for example its important for vfx guyes to render in 1920x1080, but i have no clue how much memory i would need? I know with other implementations that AVG pooling type always requires same amount of memory and would not fail once it starts, MAX pooling requires about ~25% more, and would fail on some combination of images and styles at a point it overflows

ProGamerGov commented 4 years ago

I'm not sure if such a tool would be possible. I would suggest finding a set of parameters you like, and then testing those parameters with a single 1920x1080 image. If that works, then you can use the parameters with all your images/frames.