2016-11-04 159 views

回答

0

它默認保存日誌和檢查站的/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" 

它也適用於其它例子。

0
+0

請考慮添加評論,如果你認爲這個職位可以改善。 –