因为我的训练模型在多个 GPU 上，所以在训练期间，我使用 trainer.extend 在每个时期验证我的模型，但是 chainer.training.extensions.Evaluator 只有一个设备参数，但我的模型网络参数在 2 个 GPU 上。在这种情况下如何验证？
问问题
210 次
1 回答
0
的device参数Evaluator指示应将输入数据放入哪个设备。它对模型参数没有任何作用。
如果您的模型也需要将输入数据加载到多个 GPU，Evaluator则不直接支持这种情况，因此您必须对其进行自定义。最简单的方法是设置自定义转换器功能（负责将数据复制到适当的设备）。Evaluator有关的详细信息，请参见 的文档converter。https://docs.chainer.org/en/stable/reference/generated/chainer.training.extensions.Evaluator.html
于 2017-09-14T00:04:20.633 回答