在CIFAR-10教程中暂停培训以在单GPU上运行评估

时间:2017-07-18 07:40:34

标签: python tensorflow

我正在尝试了解Tensorflow网站上的CIFAR-10教程。我可以开始运行培训,但我对评估部分感到有些困惑。它在this link中说,评估是定期运行的,但我似乎无法在代码中的任何地方找到支持它。

如何执行此操作以便暂停培训并在一定数量的步骤后定期或甚至更好地进行评估?

1 个答案:

答案 0 :(得分:1)

您必须在评估培训的同时定期运行Array.prototype.push.apply(arrryToPushTo, ArrayOfItemsToPush)(例如在另一个终端)。您可以在参数中选择运行评估的频率。

但是,正如您提供的链接中所述,您可能需要在其他设备(最好是另一个GPU)上运行它,以避免因评估在培训的同时运行而导致内存问题

相关问题