我正在使用tf.contrib.learn.Estimator
训练模型。因为我的model_dir
是Google云存储中的一个位置,上传模型需要一段时间。问题是中间结果保存在events.out.tfevents
文件中。
是否可以仅在拟合结束时写入此文件,或避免写入此文件?目前,所有的时间都花在上传这个文件上,而不是实际拟合模型。如果这只是在试衣的开始和结束,这不会是一个问题。在
estimator_full_model = tf.contrib.learn.Estimator(
model_fn=network_train.build_model_fn(),
params={'train_op': 'full_model',
'learning_rate': 0.01,
'mask_train': None},
model_dir='data',
feature_engineering_fn=train_feature_engineering_fn,
config=tf.contrib.learn.RunConfig(save_checkpoints_secs=None,
save_checkpoints_steps=100000,
log_device_placement=False))
据我所知,在培训结束时没有编写事件文件的接口,但是您可以通过将
save_summary_steps
中的save_summary_steps
增加到可接受的频率来降低编写事件文件的频率。在相关问题 更多 >
编程相关推荐