PocketFlow icon indicating copy to clipboard operation
PocketFlow copied to clipboard

TF-Plus for Multi-GPU Training

Open aakejiang opened this issue 2 years ago • 0 comments

In PocketFlow, we adopt multi-GPU training to speed-up this time-consuming training process. Our implementation is compatible with:

Horovod: a distributed training framework for TensorFlow, Keras, and PyTorch.
TF-Plus: an optimized framework for TensorFlow-based distributed training (only available within Tencent).

Is TF-Plus the private component developed by Tencent and not open-sourced yet? Are there any plan to open-source it?

aakejiang avatar Dec 28 '22 06:12 aakejiang