因为我的火车模型在多个GPU上,在训练期间,我使用trainer.extend在每个时期验证我的模型,但是chainer.training.extensions.Evaluator只有一个设备参数,但我的模型网络参数在2个GPU上。如何在这种情况下验证?
答案 0 :(得分:0)
Evaluator
的{{1}}参数表示输入数据应放在哪个设备上。它对模型参数没有任何作用。
如果您的模型也需要将输入数据加载到多个GPU,Evaluator
不直接支持此类情况,因此您必须对其进行自定义。最简单的方法是设置自定义转换器功能(负责将数据复制到适当的设备)。有关Evaluator
的详细信息,请参阅converter
的文档。 https://docs.chainer.org/en/stable/reference/generated/chainer.training.extensions.Evaluator.html