使用TensorFlow-Distributiond实现mask-r-CNN



我正在训练一个基于tensorflow和keras的mask-r-CNNN网络。我正在寻找一种减少训练时间的方法,所以我想用张量流分布式来实现它。

我已经使用 mask-r-cnn 一段时间了,但似乎我正在尝试做的事情需要我修改 mask-r-cnn 的源代码,这超出了我目前的技能。

所以,我的问题是,有人做过吗,或者类似的事情?是否有可能,或者我误解了张量流分布的使用。

谢谢。

即使没有分布式训练,Mask R-CNN 在 tensorflow 中的张量包实现也比你链接到的实现快 5 倍(也更准确(。

它还支持使用 MPI 进行分布式训练。

相关内容

  • 没有找到相关文章

最新更新