microsoft
/
caffe
зеркало из https://github.com/microsoft/caffe.git
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность

Fixed name of ArgMaxLayerParameter

This commit is contained in:
Sergio Guadarrama 2014-05-15 16:54:21 -07:00 коммит произвёл Sergey Karayev
Родитель aa57dfb84c
Коммит d6748cb994
1 изменённых файлов: 2 добавлений и 2 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

4
src/caffe/test/test_argmax_layer.cpp
Просмотреть файл

@ -51,7 +51,7 @@ TYPED_TEST(ArgMaxLayerTest, TestSetup) {
TYPED_TEST(ArgMaxLayerTest, TestSetupMaxVal) {
LayerParameter layer_param;
ArgMaxParameter* argmax_param = layer_param.mutable_argmax_param();
ArgMaxLayerParameter* argmax_param = layer_param.mutable_argmax_param();
argmax_param->set_out_max_val(true);
ThresholdLayer<TypeParam> layer(layer_param);
layer.SetUp(this->blob_bottom_vec_, &(this->blob_top_vec_));
@ -86,7 +86,7 @@ TYPED_TEST(ArgMaxLayerTest, TestCPU) {
TYPED_TEST(ArgMaxLayerTest, TestCPUMaxVal) {
LayerParameter layer_param;
Caffe::set_mode(Caffe::CPU);
ArgMaxParameter* argmax_param = layer_param.mutable_argmax_param();
ArgMaxLayerParameter* argmax_param = layer_param.mutable_argmax_param();
argmax_param->set_out_max_val(true);
ArgMaxLayer<TypeParam> layer(layer_param);
layer.SetUp(this->blob_bottom_vec_, &(this->blob_top_vec_));