tf.contrib.学习估计员避免写作events.out.tfevents事件*fi公司

2024-04-26 23:40:12 发布

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

我正在使用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))   

Tags: 文件模型nonemodelsavetfdirtrain