在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文件来恢复训练状态。