如何在chainer.training.extensions.Evaluator中使用多个GPU?

时间:2017-09-10 07:39:44

标签: chainer

因为我的火车模型在多个GPU上,在训练期间,我使用trainer.extend在每个时期验证我的模型,但是chainer.training.extensions.Evaluator只有一个设备参数,但我的模型网络参数在2个GPU上。如何在这种情况下验证?

1 个答案:

答案 0 :(得分:0)

Evaluator的{​​{1}}参数表示输入数据应放在哪个设备上。它对模型参数没有任何作用。

如果您的模型也需要将输入数据加载到多个GPU,Evaluator不直接支持此类情况,因此您必须对其进行自定义。最简单的方法是设置自定义转换器功能(负责将数据复制到适当的设备)。有关Evaluator的详细信息,请参阅converter的文档。 https://docs.chainer.org/en/stable/reference/generated/chainer.training.extensions.Evaluator.html