亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

TensorFlow 1.10+ 自定義估算器使用 train_and_evaluate 提前停止

TensorFlow 1.10+ 自定義估算器使用 train_and_evaluate 提前停止

Helenr 2021-06-30 14:47:33
假設你正在訓練的自定義tf.estimator.Estimator與tf.estimator.train_and_evaluate在類似的設置使用驗證數據集classifier = tf.estimator.Estimator(    model_fn=model_fn,    model_dir=model_dir,    params=params)train_spec = tf.estimator.TrainSpec(    input_fn = training_data_input_fn,)eval_spec = tf.estimator.EvalSpec(    input_fn = validation_data_input_fn,)tf.estimator.train_and_evaluate(    classifier,    train_spec,    eval_spec)通常,當訓練數據集的損失繼續改善而不是驗證數據集的損失繼續改善時,人們會使用驗證數據集來切斷訓練以防止過度擬合。目前tf.estimator.EvalSpec允許指定多少steps(默認為 100)來評估模型。一個人(如果可能的話不使用tf.contrib函數)如何在評估損失沒有改善的情況下指定終止訓練n(n * steps),然后將“最佳”模型/檢查點(由驗證數據集確定)保存到唯一的文件名(例如best_validation.checkpoint)
查看完整描述

1 回答

  • 1 回答
  • 0 關注
  • 172 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號