Tencent/PocketFlow

TF-Plus for Multi-GPU Training

aakejiang opened this issue · 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?