Skip to content
This repository was archived by the owner on May 24, 2018. It is now read-only.

Commit 6171751

Browse files
committed
final fix
1 parent 72393c8 commit 6171751

File tree

3 files changed

+8
-8
lines changed

3 files changed

+8
-8
lines changed

cxxnet/cxxnet_main.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -194,11 +194,11 @@ namespace cxxnet{
194194
}
195195
if( !strcmp( name, "iter" ) && !strcmp( val, "end" ) ){
196196
utils::Assert( flag != 0, "wrong configuration file" );
197-
if( flag == 1 ){
197+
if( flag == 1 && task != "pred" ){
198198
utils::Assert( itr_train == NULL, "can only have one data" );
199199
itr_train = cxxnet::CreateIterator( itcfg );
200200
}
201-
if( flag == 2 ){
201+
if( flag == 2 && task != "pred" ){
202202
itr_evals.push_back( cxxnet::CreateIterator( itcfg ) );
203203
eval_names.push_back( evname );
204204
}

example/MNIST/MNIST.conf

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,9 +36,9 @@ num_round = 15
3636
train_eval = 1
3737
random_type = gaussian
3838
## learning parameters
39-
lr:eta = 0.1
40-
lr:momentum = 0.9
41-
lr:wd = 0.0
39+
eta = 0.1
40+
momentum = 0.9
41+
wd = 0.0
4242
# evaluation metric
4343
metric = error
4444
# end of config

example/MNIST/MNIST_CONV.conf

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -50,9 +50,9 @@ num_round = 15
5050
train_eval = 1
5151
random_type = gaussian
5252
## learning parameters
53-
lr:eta = 0.1
54-
lr:momentum = 0.9
55-
lr:wd = 0.0
53+
eta = 0.1
54+
momentum = 0.9
55+
wd = 0.0
5656
# evaluation metric
5757
metric = error
5858
eval_train = 1

0 commit comments

Comments
 (0)