diluvian.third_party package

Submodules

diluvian.third_party.multi_gpu module

Parallelize training a Keras model across multiple GPUs.

Originally by Alex Kouzemtchenko, taken from: https://github.com/kuza55/keras-extras/utils/multi_gpu.py

License: Apache

diluvian.third_party.multi_gpu.make_parallel(model, gpu_count)

Module contents