2017-05-22 64 views
0

我喜歡爲我培訓的每個時代提供一組新的實例。唯一的辦法就是重置每個紀元的隊列。但是,一旦我們將隊列標記爲圖形,則無法編輯其內容。你知道任何解決方案或其他想法嗎?重置每個紀元後Tensorflow輸入隊列的內容?

我使用string_input_producer排隊文件路徑,然後動態讀取它們並將它們解碼爲TF圖形中的批處理。

回答

1

tf.contrib.data API的設計正好解決了這樣的問題,我建議您使用它來代替隊列。