Commit 733b7f32 authored by Evgeny Kusmenko's avatar Evgeny Kusmenko
Browse files

Merge branch 'context_value_bugfix' into 'master'

Bugfix for correct context (CPU/GPU) value

See merge request !7
parents 226d1816 f824022e
Pipeline #66826 passed with stages
in 2 minutes and 24 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