From 47f24169ba48b1079958c663abfe691a6c1d0ada Mon Sep 17 00:00:00 2001 From: ian Date: Wed, 21 Sep 2016 21:04:12 +0800 Subject: [PATCH] fix a layer name in googlenetv2 --- .../scala/com/intel/analytics/sparkdl/models/GoogleNet.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dl/src/main/scala/com/intel/analytics/sparkdl/models/GoogleNet.scala b/dl/src/main/scala/com/intel/analytics/sparkdl/models/GoogleNet.scala index 12c1a41f100..cec63aefce5 100644 --- a/dl/src/main/scala/com/intel/analytics/sparkdl/models/GoogleNet.scala +++ b/dl/src/main/scala/com/intel/analytics/sparkdl/models/GoogleNet.scala @@ -232,7 +232,7 @@ object GoogleNet_v2 { val conv3 = new Sequential[D] conv3.add(new SpatialConvolution[D](inputSize, config[Table](2)(1), 1, 1, 1, 1) - .setName(namePrefix + "3x3_s2")) + .setName(namePrefix + "3x3_reduce")) conv3.add(new SpatialBatchNormalization(config[Table](2)(1), 1e-3) .setName(namePrefix + "3x3_reduce/bn")) conv3.add(new ReLU[D](true). setName(namePrefix + "3x3_reduce/bn/sc/relu"))