caffe2 cpu mode

adedd419 · nilsfreyer · 072662a1 · adedd419
Commit adedd419 authored Mar 07, 2019 by nilsfreyer
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/test/resources/models/cNNCalculator/Network.cnnt src/test/resources/models/cNNCalculator/Network.cnnt +1 -1

No files found.
--- a/src/test/resources/models/cNNCalculator/Network.cnnt
+++ b/src/test/resources/models/cNNCalculator/Network.cnnt
 configuration Network{
     num_epoch:11
     batch_size:64
-     context:gpu
+     context:cpu
     eval_metric:accuracy
     optimizer:adam{
         learning_rate:0.001