Commit adedd419 authored by nilsfreyer's avatar nilsfreyer

caffe2 cpu mode

parent 072662a1
configuration Network{ configuration Network{
num_epoch:11 num_epoch:11
batch_size:64 batch_size:64
context:gpu context:cpu
eval_metric:accuracy eval_metric:accuracy
optimizer:adam{ optimizer:adam{
learning_rate:0.001 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