Closed ProGamerGov closed 4 years ago
For example this put the first layer of the model on the CPU, while every other layer is on the GPU::
python3 neural_style.py -gpu c,0 -image_size 256 -multidevice_strategy 1
And this will put the first and last layers of a model on the CPU, while every other layer is on the GPU:
python3 neural_style.py -gpu c,0,c -image_size 256 -multidevice_strategy 1,34
For example this put the first layer of the model on the CPU, while every other layer is on the GPU::
And this will put the first and last layers of a model on the CPU, while every other layer is on the GPU: