From 9faf20876a19f70cecd46cd4ddf7c9bdd058e344 Mon Sep 17 00:00:00 2001
From: LordMathis <matus@namesny.com>
Date: Thu, 30 May 2024 15:21:01 +0200
Subject: [PATCH] Fix branch names

---
 examples/inception_v3/inception_v3.cpp | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/examples/inception_v3/inception_v3.cpp b/examples/inception_v3/inception_v3.cpp
index a19db62..f8deea8 100644
--- a/examples/inception_v3/inception_v3.cpp
+++ b/examples/inception_v3/inception_v3.cpp
@@ -219,7 +219,7 @@ class InceptionB : public CUDANet::Module {
         // Branch 3x3
         branch3x3 = new BasicConv2d(
             inputShape, inputChannels, 384, {3, 3}, {2, 2}, {0, 0},
-            prefix + ".branch1x1"
+            prefix + ".branch3x3"
         );
         addLayer("", branch3x3);
 
@@ -230,7 +230,7 @@ class InceptionB : public CUDANet::Module {
         );
         addLayer("", branch3x3dbl_1);
         branch3x3dbl_2 = new BasicConv2d(
-            branch3x3dbl_1->getOutputDims(), 96, 96, {3, 3}, {1, 1}, {1, 1},
+            branch3x3dbl_1->getOutputDims(), 64, 96, {3, 3}, {1, 1}, {1, 1},
             prefix + ".branch3x3dbl_2"
         );
         addLayer("", branch3x3dbl_2);
@@ -485,7 +485,7 @@ class InceptionD : public CUDANet::Module {
         // Branch 3x3
         branch3x3_1 = new BasicConv2d(
             inputShape, inputChannels, 192, {1, 1}, {1, 1}, {0, 0},
-            prefix + ".branch3x3"
+            prefix + ".branch3x3_1"
         );
         addLayer("", branch3x3_1);
         branch3x3_2 = new BasicConv2d(