Skip to content

v0.2 MultiGPU train

Latest
Compare
Choose a tag to compare
@T-Almeida T-Almeida released this 07 Jun 11:59
· 11 commits to master since this release

Adding multiGPU training with the integration of the horovod library, which handles the communication between the GPU or machines.

Adding a dockerfile for installing the horovod since it can be a hassle to install horovod with the correct dependencies

This version also denotes a compatibility checkpoint with the BioASQ2022