初始化MonitoredTrainingSession中的变量子集

2024-04-26 07:26:59 发布

您现在位置:Python中文网/ 问答频道 /正文

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

但是,我不知道如何使用MonitoredTrainingSessionMonitoredTrainingSession对我来说有很多好处,我不想放弃它,如果有办法让它与MonitoredTrainingSession一起工作,我就切换到tf.Session


Tags: 代码sessiontf检查点子集权重saver办法