在python wrraper caffe上訓練網絡？

Question

我想用python界面訓練caffe網絡。 背後的主要原因是我使用多個Tbs數據的多維輸入，我不想將所有這些轉換爲LMDB並進行訓練。

我已經找到了這個one answer on stack overflow.

但他的加載這個完整的一次數據，並初始化權重。

我想將數據加載到numpy然後傳遞給caffe。

然後每1000次迭代一次，將caffemodel的權重保存到.caffemodel文件中。

print_network() get_accuracy() & load_data()非常有用。並給我一個很好的內心。

Answer 1

除了使用PythonLayer，有一兩件事你可以做的是在時間需要然而，許多迭代要經過一個批量數據的使用後，使用solver.net.set_input_arrays(your_data)層MemoryData和飼料中的每個數據批次。

請記住，您始終可以通過使用快照中的.solverstate文件來恢復訓練狀態。