在Caffe的prototxt文件中,TRAIN和TEST阶段的作用是什么?

我是Caffe的新手,谢谢大家!

https://github.com/BVLC/caffe/blob/master/src/caffe/proto/caffe.proto

我看到一个未注释的枚举变量phase,它有两个选项TRAIN和TEST。

enum Phase {   TRAIN = 0;   TEST = 1;}

它们是如何工作的?我最近看到一个模型也有这两个阶段。.prototxt文件看起来像这样:

name: "CIFAR10_full"layer {  name: "cifar"  type: "Data"  top: "data"  top: "label"  data_param {    source: "CIFAR-10/cifar10_train_lmdb"    backend: LMDB    batch_size: 200  }  transform_param {    mirror: true  }  include: { phase: TRAIN }}layer {  name: "cifar"  type: "Data"  top: "data"  top: "label"  data_param {    source: "CIFAR-10/cifar10_test_lmdb"    backend: LMDB    batch_size: 100  }  transform_param {    mirror: false  }  include: { phase: TEST }}

我可以从TRAIN阶段切换到TEST阶段吗?切换在哪里进行?


回答:

在训练过程中(即执行$CAFFE_ROOT/tools/caffe train [...]),Caffe可以在训练阶段和测试阶段之间交替:在训练阶段,参数会发生变化,而在测试阶段,参数是固定的,模型仅运行前馈示例以估计当前的模型性能。
使用两个不同的数据集进行训练和测试是非常自然的,这就是为什么你会使用不同的phase值。

你可以在这里阅读更多关于训练/测试迭代的信息。

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注