Skip to content

Commit cf189c5

Browse files
committed
Adjust tests
1 parent d3e8492 commit cf189c5

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

test/single_layer/test_dropoutlayer.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ TEST(DropOutLayer, IncompatibleInput) {
1717
}
1818

1919
TEST(DropOutLayer, dropoutlayer_int) {
20-
DropOutLayer layer(1);
20+
DropOutLayer layer(1, true);
2121
Shape sh({2, 2});
2222
Tensor input = make_tensor<int>({1, -1, 2, -2}, sh);
2323
Tensor output;
@@ -47,7 +47,7 @@ TEST(DropOutLayer, dropoutlayer_float) {
4747
}
4848

4949
TEST(DropOutLayer, dropoutlayer_float_50proc) {
50-
DropOutLayer layer(0.5);
50+
DropOutLayer layer(0.5, true);
5151
Shape sh({10, 10});
5252
std::vector<float> a(100, static_cast<float>(0.01));
5353
Tensor input = make_tensor<float>(a, sh);
@@ -60,7 +60,7 @@ TEST(DropOutLayer, dropoutlayer_float_50proc) {
6060
}
6161

6262
TEST(DropOutLayer, dropoutlayer_float_30proc) {
63-
DropOutLayer layer(0.3);
63+
DropOutLayer layer(0.3, true);
6464
Shape sh({10, 10});
6565
std::vector<float> a(100, static_cast<float>(0.01));
6666
Tensor input = make_tensor<float>(a, sh);
@@ -73,7 +73,7 @@ TEST(DropOutLayer, dropoutlayer_float_30proc) {
7373
}
7474

7575
TEST(DropOutLayer, dropoutlayer_float_70proc) {
76-
DropOutLayer layer(0.7);
76+
DropOutLayer layer(0.7, true);
7777
Shape sh({10, 10});
7878
std::vector<float> a(100, static_cast<float>(0.01));
7979
Tensor input = make_tensor<float>(a, sh);

0 commit comments

Comments
 (0)