它默認保存日誌和檢查站的/tmp/
文件夾。 權重包含在檢查點文件中。
正如您可以在eval和train文件中看到的那樣,它將檢查點目錄作爲參數。
cifar10_train.py:
tf.app.flags.DEFINE_string('train_dir', '/tmp/cifar10_train',
"""Directory where to write event logs """
"""and checkpoint.""")
cifar10_eval.py:
tf.app.flags.DEFINE_string('eval_dir', '/tmp/cifar10_eval',
"""Directory where to write event logs.""")
tf.app.flags.DEFINE_string('eval_data', 'test',
"""Either 'test' or 'train_eval'.""")
tf.app.flags.DEFINE_string('checkpoint_dir', '/tmp/cifar10_train',
"""Directory where to read model checkpoints.""")
你可以調用這些腳本與這些自定義值。對於我的使用Inception的項目,我必須改變它,因爲主硬盤沒有足夠的空間來容納最初創建的瓶頸。
由於/tmp/
文件夾不是持久性的,因此您可能會丟失訓練數據,因此明確設置這些值可能是一種很好的做法。
以下代碼會將訓練數據保存到自定義文件夾中。
python cifar10_train.py --train_dir="/home/username/train_folder"
,然後評估:
python cifar10_eval.py --checkpoint_dir="/home/username/train_folder"
它也適用於其它例子。
請考慮添加評論,如果你認爲這個職位可以改善。 –