How to do some GPU training with Keras?

I want my model to run on several GPU sharing options, but with different batches of data.

Can I do something similar using model.fit()? Is there any other alternative?

+6
source share
2 answers

try using the make_parallel function in: https://github.com/kuza55/keras-extras/blob/master/utils/multi_gpu.py (it will work only with the backend tensor).

+1
source
0

All Articles