即使找到检查点,Tensorflow warm_start

时间:2018-08-02 15:24:57

标签: python-3.x tensorflow tensorflow-estimator transfer-learning

我注意到,如果模型目录中是否存在检查点,则标记为warm_start的变量将开始热启动。如果已经有检查点,有没有办法不热启动?还是我将这些变量从训练中排除掉与事实有什么关系?

我正在使用自定义估算器中的warm_start来初始化inception_v4,从中我将最后一层用作模型输入。

 id  Place Year_2000 Year_2010 Year_2020 Year
 AL_Scen1_10mm.land.csv Putnam 0 0.17 0.19 
 AL_Scen2_5mm.ocean.csv Boykin 0.1 0.3 0.5

此刻我也将他们排除在训练范围之外

ws = tf.estimator.WarmStartSettings(ckpt_to_initialize_from=self.warm_start_dir,
                                             vars_to_warm_start='.*InceptionV4.*')

但是我也想不排除它们,这将是一个更大的问题。

0 个答案:

没有答案