final fix

dmlc · Apr 8, 2014 · 6171751 · 6171751
1 parent 72393c8
commit 6171751
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/cxxnet/cxxnet_main.cpp b/cxxnet/cxxnet_main.cpp
@@ -194,11 +194,11 @@ namespace cxxnet{
                 }
                 if( !strcmp( name, "iter" ) && !strcmp( val, "end" ) ){
                     utils::Assert( flag != 0, "wrong configuration file" );
-                    if( flag == 1 ){
+                    if( flag == 1 && task != "pred" ){
                         utils::Assert( itr_train == NULL, "can only have one data" );
                         itr_train = cxxnet::CreateIterator( itcfg );
                     }
-                    if( flag == 2 ){
+                    if( flag == 2 && task != "pred" ){
                         itr_evals.push_back( cxxnet::CreateIterator( itcfg ) );
                         eval_names.push_back( evname );
                     }

diff --git a/example/MNIST/MNIST.conf b/example/MNIST/MNIST.conf
@@ -36,9 +36,9 @@ num_round = 15
 train_eval = 1
 random_type = gaussian
 ## learning parameters
-lr:eta = 0.1
-lr:momentum = 0.9
-lr:wd  = 0.0
+eta = 0.1
+momentum = 0.9
+wd  = 0.0
 # evaluation metric
 metric = error
 # end of config
diff --git a/example/MNIST/MNIST_CONV.conf b/example/MNIST/MNIST_CONV.conf
@@ -50,9 +50,9 @@ num_round = 15
 train_eval = 1
 random_type = gaussian
 ## learning parameters
-lr:eta = 0.1
-lr:momentum = 0.9
-lr:wd  = 0.0
+eta = 0.1
+momentum = 0.9
+wd  = 0.0
 # evaluation metric
 metric = error
 eval_train = 1