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 @@ ...@@ -15,8 +15,8 @@
<properties> <properties>
<!-- .. SE-Libraries .................................................. --> <!-- .. SE-Libraries .................................................. -->
<CNNArch.version>0.2.1-SNAPSHOT</CNNArch.version> <CNNArch.version>0.2.3-SNAPSHOT</CNNArch.version>
<CNNTrain.version>0.2.2-SNAPSHOT</CNNTrain.version> <CNNTrain.version>0.2.4-SNAPSHOT</CNNTrain.version>
<!-- .. Libraries .................................................. --> <!-- .. Libraries .................................................. -->
<guava.version>18.0</guava.version> <guava.version>18.0</guava.version>
......
${element.name} = mx.symbol.Pooling(data=${element.inputs[0]}, ${element.name} = mx.symbol.Pooling(data=${element.inputs[0]},
global_pool=True, global_pool=True,
kernel=(1,1), kernel=(1,1),
pool_type=${element.poolType}, pool_type="${element.poolType}",
name="${element.name}") name="${element.name}")
<#include "OutputShape.ftl"> <#include "OutputShape.ftl">
\ No newline at end of file
...@@ -8,7 +8,7 @@ ...@@ -8,7 +8,7 @@
</#if> </#if>
${element.name} = mx.symbol.Pooling(data=${input}, ${element.name} = mx.symbol.Pooling(data=${input},
kernel=(${tc.join(element.kernel, ",")}), kernel=(${tc.join(element.kernel, ",")}),
pool_type=${element.poolType}, pool_type="${element.poolType}",
stride=(${tc.join(element.stride, ",")}), stride=(${tc.join(element.stride, ",")}),
name="${element.name}") name="${element.name}")
<#include "OutputShape.ftl"> <#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){ ...@@ -28,4 +28,4 @@ architecture VGG16(img_height=224, img_width=224, img_channels=3, classes=1000){
FullyConnected(units=classes) -> FullyConnected(units=classes) ->
Softmax() -> Softmax() ->
predictions 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