Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 01:34:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rework inception block tests

Browse Source
This commit is contained in:
Matúš Námešný
2024-06-05 21:58:33 +02:00
parent a54ffa8b20
commit 1136ca452f
12 changed files with 416 additions and 104738 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/inception_v3/tests/test_basic_conv2d.cpp
View File

@@ -12,7 +12,7 @@ class BasicConv2dTest : public ::testing::Test {
shape2d kernelSize;
shape2d stride;
shape2d padding;
std::string prefix = "test";
std::string prefix = "test.";
float *d_input;
float *d_output;
@@ -46,7 +46,7 @@ class BasicConv2dTest : public ::testing::Test {
std::pair<std::string, CUDANet::Layers::SequentialLayer *> layerPair =
basic_conv2d->getLayers()[0];
ASSERT_EQ(layerPair.first, prefix + ".conv");
ASSERT_EQ(layerPair.first, prefix + "conv");
CUDANet::Layers::Conv2d *conv =
dynamic_cast<CUDANet::Layers::Conv2d *>(layerPair.second);
@@ -60,7 +60,7 @@ class BasicConv2dTest : public ::testing::Test {
EXPECT_EQ(cudaStatus, cudaSuccess);
layerPair = basic_conv2d->getLayers()[1];
ASSERT_EQ(layerPair.first, prefix + ".bn");
ASSERT_EQ(layerPair.first, prefix + "bn");
CUDANet::Layers::BatchNorm2d *bn =
dynamic_cast<CUDANet::Layers::BatchNorm2d *>(layerPair.second);