Switched to cpu training

parent a7e2dbc6
......@@ -2,7 +2,7 @@ configuration Network{
num_epoch:5
batch_size:100
normalize:true
context:gpu
context:cpu
load_checkpoint:false
optimizer:sgd{
learning_rate:0.2
......
configuration Network{
num_epoch:11
batch_size:64
context:gpu
context:cpu
eval_metric:accuracy
optimizer:adam{
learning_rate:0.001
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment