Train test on cpu

parent 376a7773
Pipeline #109593 failed with stage
in 3 minutes and 38 seconds
......@@ -2,7 +2,7 @@ configuration CifarNetwork{
num_epoch:10
batch_size:5
normalize:true
context:gpu
context:cpu
load_checkpoint:false
optimizer:adam{
learning_rate:0.01
......
......@@ -2,7 +2,7 @@ configuration CifarNetwork{
num_epoch:10
batch_size:5
normalize:true
context:gpu
context:cpu
load_checkpoint:false
optimizer:adam{
learning_rate:0.01
......
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