Skip to content
Snippets Groups Projects
Commit b71bd9c2 authored by Benjamin Halimi's avatar Benjamin Halimi
Browse files

add the training flag in the unit tests

parent 8560702e
No related branches found
No related tags found
1 merge request!263Add the BatchNorm train/test flag support
Pipeline #60273 passed
...@@ -352,11 +352,11 @@ TEST_CASE("[core/graph] Matching") { ...@@ -352,11 +352,11 @@ TEST_CASE("[core/graph] Matching") {
auto g2 = Sequential({ auto g2 = Sequential({
Producer({16, 3, 512, 512}, "dataProvider"), Producer({16, 3, 512, 512}, "dataProvider"),
Conv(3, 4, {5, 5}, "conv1"), Conv(3, 4, {5, 5}, "conv1"),
BatchNorm<2>(4, 1.0e-5, 0.1, "bn1"), BatchNorm<2>(4, 1.0e-5, 0.1, 0, "bn1"),
Conv(4, 4, {5, 5}, "conv2"), Conv(4, 4, {5, 5}, "conv2"),
ReLU("relu2"), ReLU("relu2"),
Conv(4, 4, {5, 5}, "conv3"), Conv(4, 4, {5, 5}, "conv3"),
BatchNorm<2>(4, 1.0e-5, 0.1, "bn3"), BatchNorm<2>(4, 1.0e-5, 0.1, 0, "bn3"),
FC(4, 4, false, "fc1"), FC(4, 4, false, "fc1"),
FC(4, 4, false, "fc2"), FC(4, 4, false, "fc2"),
FC(4, 4, false, "fc3"), FC(4, 4, false, "fc3"),
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment