1. 程式人生 > >caffe訓練資料中出現[data_layer.cpp:73] Restarting data prefetching from start. 的問題

caffe訓練資料中出現[data_layer.cpp:73] Restarting data prefetching from start. 的問題

        一開始在網上查了許多的,有人回答說是之前的一些prototxt檔案引數設定有問題,還有人說是batch_size相對於整個資料集太大的問題。

        我一開始也以為這是個warning提示,會對訓練結果產生影響,但是穩步上升的accuracy,又覺得是沒問題的。

        於是我現在覺得,這個提示,僅僅是個提示,對訓練結果是沒有任何影響的。

        我的看法是,這個提示僅僅說明遍歷了一次訓練集的訓練輸入。因此我之前的訓練集有4400個,batch_size設為44,test_iter設定為100,而且設定了display為100,因此每訓練100次,就遍歷完整個訓練集,便會提示一次,restarting,同時,也是我選擇display一次的時機。

        為了驗證這個想法,我將display設為400,即每400次才打印一次訓練的loss等結果,想法正確的話,應該會每顯示4次restarting後列印一次。

        我覺得這就驗證成功了,當然,有時間還是看原始碼比較準確,可惜最近忙,沒時間,算了,有時間再看看。