bert 多GPU Horovod

bfnvny8b  于 3个月前  发布在  其他
关注(0)|答案(2)|浏览(48)

在使用horovod进行多GPU训练时,examples/sec的值是针对一个GPU还是所有GPU?我发现多GPU的值低于单GPU。谢谢。

kmb7vmvb

kmb7vmvb1#

你在哪里看到的?

7vux5j2d

7vux5j2d2#

@tianuna,你在哪里看到的?
这是单个GPU的结果。

这是使用horovod的4个GPU的结果。我认为每秒训练样本数应该是44*4=176。但我找不到定义这个的代码。

相关问题