在训练Keras模型时,是否有办法让GPU在设定的时间间隔之间进行小休息(睡眠)?

时间:2019-01-02 01:57:30

标签: python tensorflow machine-learning keras artificial-intelligence

我的Keras模型当前正在查看大量数据,我个人不愿意让我的GPU达到85度...在一定时间间隔内是否有办法告诉我的GPU休息一下?

我知道我可以将整个过程分为多个训练周期,但是因为我在RNN模型的回调中使用ReduceLROnPlateau,所以我仍然希望整个训练过程可以在一个训练周期中完成,而GPU占用的空间很小。休息以延长培训时间,而对我的个人硬件的风险也较小。

(由于这只是一个普遍的问题,所以不添加代码。)

0 个答案:

没有答案