Commit f824022e authored by Svetlana Pavlitskaya's avatar Svetlana Pavlitskaya
Browse files

Now correct value of context is generated.

parent 226d1816
Pipeline #66617 passed with stages
in 2 minutes and 51 seconds
......@@ -161,7 +161,7 @@ public class CNNTrainSymbolTableCreator extends CNNTrainSymbolTableCreatorTOP {
value.setValue(Context.CPU);
}
else {
value.setValue(Context.CPU);
value.setValue(Context.GPU);
}
entry.setValue(value);
addToScopeAndLinkWithNode(entry, node);
......
......@@ -3,7 +3,7 @@ configuration FullConfig{
batch_size : 100
load_checkpoint : true
eval_metric : mse
context : cpu
context : gpu
normalize : true
optimizer : rmsprop{
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