Updated pom: cnn-arch version changed to 0.2.3-SNAPSHOT and cnn-train version...

Updated pom: cnn-arch version changed to 0.2.3-SNAPSHOT and cnn-train version changed to 0.2.4-SNAPSHOT.
Minor changes in pooling layers.
parent 839e7cc0
Pipeline #69754 failed with stages
......@@ -15,8 +15,8 @@
<properties>
<!-- .. SE-Libraries .................................................. -->
<CNNArch.version>0.2.1-SNAPSHOT</CNNArch.version>
<CNNTrain.version>0.2.2-SNAPSHOT</CNNTrain.version>
<CNNArch.version>0.2.3-SNAPSHOT</CNNArch.version>
<CNNTrain.version>0.2.4-SNAPSHOT</CNNTrain.version>
<!-- .. Libraries .................................................. -->
<guava.version>18.0</guava.version>
......
${element.name} = mx.symbol.Pooling(data=${element.inputs[0]},
global_pool=True,
kernel=(1,1),
pool_type=${element.poolType},
pool_type="${element.poolType}",
name="${element.name}")
<#include "OutputShape.ftl">
\ No newline at end of file
......@@ -8,7 +8,7 @@
</#if>
${element.name} = mx.symbol.Pooling(data=${input},
kernel=(${tc.join(element.kernel, ",")}),
pool_type=${element.poolType},
pool_type="${element.poolType}",
stride=(${tc.join(element.stride, ",")}),
name="${element.name}")
<#include "OutputShape.ftl">
\ No newline at end of file
......@@ -28,4 +28,4 @@ architecture VGG16(img_height=224, img_width=224, img_channels=3, classes=1000){
FullyConnected(units=classes) ->
Softmax() ->
predictions
}
}
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment