在MonitoredTrainingSession中初始化变量子集

时间:2019-01-08 00:08:25

标签: python tensorflow gpu

我在训练代码中使用MonitoredTrainingSession,我想从检查点初始化权重子集,并将其余未分配的变量初始化为随机值。我知道如何使用tf.session来执行此操作,并首先初始化所有变量,然后使用tf.saver从检查点读取变量。

但是,我不知道如何使用MonitoredTrainingSessionMonitoredTrainingSession对我有很多好处,如果有办法使它与tf.Session一起使用,我不愿意放弃并切换到MonitoredTrainingSession

1 个答案:

答案 0 :(得分:0)

您可以尝试:

import copy
m_bak = copy.deepcopy(MonitoredTrainingSession)