在使用horovod进行多GPU训练时,examples/sec的值是针对一个GPU还是所有GPU?我发现多GPU的值低于单GPU。谢谢。
kmb7vmvb1#
你在哪里看到的?
7vux5j2d2#
@tianuna,你在哪里看到的?这是单个GPU的结果。
这是使用horovod的4个GPU的结果。我认为每秒训练样本数应该是44*4=176。但我找不到定义这个的代码。
2条答案
按热度按时间kmb7vmvb1#
你在哪里看到的?
7vux5j2d2#
@tianuna,你在哪里看到的?
这是单个GPU的结果。
这是使用horovod的4个GPU的结果。我认为每秒训练样本数应该是44*4=176。但我找不到定义这个的代码。