eridgd / WCT-TF

TensorFlow/Keras implementation of "Universal Style Transfer via Feature Transforms" from https://arxiv.org/abs/1705.08086
MIT License
319 stars 79 forks source link

Is there a way to use multiple GPUs to evaluate a style? #18

Open t-eialka opened 6 years ago

t-eialka commented 6 years ago

Is there a way to stylize an image using multiple GPUs to speed up the process?

eridgd commented 6 years ago

I don't think that's possible with this algorithm, I'm not seeing obvious places in the graph where it would be possible to parallelize ops.

kobykotiv commented 6 years ago

I think you can run the algorithm twice. With one command line set stylize.py to one gpu --Device gpu:0, and the the other one(s) --Device gpu:1