CC4.0训练流程图

NO

配置层layer.setup

installRegister安装注册器

数据输出layer.top

缓存是否有可用Batch

INSTALL_LAYER()注册层，关联MNistDataLayer

所有layer

加载test_net: train.prototxt参数

等待10ms

创建层layer.create

数据层加载批次MNistDataLayer::loadBatch

加载Train网络

Main

启动加载器线程startWatcher

最大次数？iter max_iter

加载一批图片loadBatch

注册回调函数等registerLayerFunctionregisterLayerForwardFunction

构建数据加载器setupBatch

没有空Batch

保存模型和快照训练结束

CustomLayerForward

YES

是否存在空Batch

MNistDataLayer::Forward

CNN backward

根据net参数构建网络

从Batch缓存中获取一个Watcher得到反应

符合测试条件？test_interval

CNN forward

solver-solve()开始求解训练

记录MNistDataLayer的create和release（由SETUP_LAYERFUNC定义）

获取一个Batch

DataLayer::pullBatch

DataLayer::Forward

存在空Batch

准备数据prepaperData

输出测试结果

输出loss等信息

loadSolverFromPrototxt加载求解器solver.prototxt

加载Test网络

TEST_NET::Forwardtest_iter

加载train_net: train.prototxt参数