Skip to content

Commit 96c140b

Browse files
committed
[ONN] Add batch norm layer to nerlplanner
1 parent bfd5c09 commit 96c140b

File tree

7 files changed

+21
-19
lines changed

7 files changed

+21
-19
lines changed

src_cpp/common/worker_definitions_ag.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
namespace nerlnet {
77

8-
enum LayerTypeEnum{LAYER_TYPE_DEFAULT=0,LAYER_TYPE_SCALING=1,LAYER_TYPE_CONV=2,LAYER_TYPE_PERCEPTRON=3,LAYER_TYPE_POOLING=4,LAYER_TYPE_PROBABILISTIC=5,LAYER_TYPE_LSTM=6,LAYER_TYPE_RECCURRENT=7,LAYER_TYPE_UNSCALING=8,LAYER_TYPE_FLATTEN=9,LAYER_TYPE_BOUNDING=10};
8+
enum LayerTypeEnum{LAYER_TYPE_DEFAULT=0,LAYER_TYPE_SCALING=1,LAYER_TYPE_CONV=2,LAYER_TYPE_PERCEPTRON=3,LAYER_TYPE_POOLING=4,LAYER_TYPE_PROBABILISTIC=5,LAYER_TYPE_BATCHNORMALIZATION=6,LAYER_TYPE_LSTM=7,LAYER_TYPE_RECCURRENT=8,LAYER_TYPE_UNSCALING=9,LAYER_TYPE_FLATTEN=10,LAYER_TYPE_BOUNDING=11};
99
enum ProbabilisticActivationEnum{PROBABILISTIC_ACTIVATION_BINARY=1,PROBABILISTIC_ACTIVATION_LOGISTIC=2,PROBABILISTIC_ACTIVATION_COMPETITIVE=3,PROBABILISTIC_ACTIVATION_SOFTMAX=4};
1010
enum ScalingEnum{SCALING_NONE=1,SCALING_MINMAX=2,SCALING_MEANSTD=3,SCALING_STD=4,SCALING_LOG=5};
1111
enum BoundingEnum{BOUNDING_NONE=1,BOUNDING_BOUNDING=2};

src_erl/NerlnetApp/src/Bridge/layers_types_ag.hrl

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,9 @@
77
-define(LAYERS_TYPE_PERCEPTRON_IDX,"3").
88
-define(LAYERS_TYPE_POOLING_IDX,"4").
99
-define(LAYERS_TYPE_PROBABILISTIC_IDX,"5").
10-
-define(LAYERS_TYPE_LSTM_IDX,"6").
11-
-define(LAYERS_TYPE_RECCURRENT_IDX,"7").
12-
-define(LAYERS_TYPE_UNSCALING_IDX,"8").
13-
-define(LAYERS_TYPE_FLATTEN_IDX,"9").
14-
-define(LAYERS_TYPE_BOUNDING_IDX,"10").
10+
-define(LAYERS_TYPE_BATCHNORMALIZATION_IDX,"6").
11+
-define(LAYERS_TYPE_LSTM_IDX,"7").
12+
-define(LAYERS_TYPE_RECCURRENT_IDX,"8").
13+
-define(LAYERS_TYPE_UNSCALING_IDX,"9").
14+
-define(LAYERS_TYPE_FLATTEN_IDX,"10").
15+
-define(LAYERS_TYPE_BOUNDING_IDX,"11").

src_erl/NerlnetApp/src/Bridge/neural_networks_testing_models.hrl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
_ModelTypeCNN = "0",
2020
_ModelArgsCNN = "",
2121
_LayersSizesCNN = "28x28x1k5x5x1x6p0s1t1,28x28x6k2x2p0s2,14x14x6k4x4x6x12p0s1t0,1,32,10",
22-
_LayersTypesCNN = "2,4,2,9,3,5",
22+
_LayersTypesCNN = "2,4,2,10,3,5",
2323
_LayersFunctionalityCodesCNN = "6,2,6,6,6,4", % change scaler functionality to 6 to check exception handling
2424
_LearningRateCNN = "0.01",
2525
_EpochsCNN = "50",
@@ -33,7 +33,7 @@
3333
_ModelTypeAEC = "9",
3434
_ModelArgsAEC = "",
3535
_LayersSizesAEC = "32,16,8,4,8,16,32,32", % last layer (perceptron) should be the same as the input layer , followed by bounding layer
36-
_LayersTypesAEC = "1,3,3,3,3,3,3,10",
36+
_LayersTypesAEC = "1,3,3,3,3,3,3,11",
3737
_LayersFunctionalityCodesAEC = "1,11,11,11,11,11,11,1",
3838
_LearningRateAEC = "0.01",
3939
_EpochsAEC = "50",
@@ -47,7 +47,7 @@
4747
_ModelTypeAE = "8",
4848
_ModelArgsAE = "",
4949
_LayersSizesAE = "32,16,8,4,8,16,32,32", % last layer (perceptron) should be the same as the input layer , followed by bounding layer
50-
_LayersTypesAE = "1,3,3,3,3,3,3,10",
50+
_LayersTypesAE = "1,3,3,3,3,3,3,11",
5151
_LayersFunctionalityCodesAE = "1,11,11,11,11,11,11,1",
5252
_LearningRateAE = "0.01",
5353
_EpochsAE = "50",

src_erl/NerlnetApp/src/nerlnetApp.app.src

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
{application, nerlnetApp,
22
[{description, "Nerlnet OTP Application"},
3-
{vsn, "1.4.0"},
3+
{vsn, "1.5.1"},
44
{registered, []},
55
{mod, {nerlnetApp_app, []}},
66
{applications,

src_erl/NerlnetApp/src/nerlnetApp_app.erl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,8 @@
2020
-behaviour(application).
2121
-include("nerl_tools.hrl").
2222

23-
-define(NERLNET_APP_VERSION, "1.5.0").
24-
-define(NERLPLANNER_TESTED_VERSION,"1.0.2").
23+
-define(NERLNET_APP_VERSION, "1.5.1").
24+
-define(NERLPLANNER_TESTED_VERSION,"1.0.3").
2525

2626
-export([start/2, stop/1]).
2727

src_py/nerlPlanner/Definitions.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
import subprocess
22
from logger import *
33

4-
VERSION = "1.0.2"
5-
NERLNET_VERSION_TESTED_WITH = "1.5.0"
4+
VERSION = "1.0.3"
5+
NERLNET_VERSION_TESTED_WITH = "1.5.1"
66
NERLNET_TMP_PATH = "/tmp/nerlnet"
77
NERLNET_GRAPHVIZ_OUTPUT_DIR = f"{NERLNET_TMP_PATH}/nerlplanner"
88
NERLNET_GLOBAL_PATH = "/usr/local/lib/nerlnet-lib/NErlNet"

src_py/nerlPlanner/JsonElementWorkerDefinitions.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,12 @@
1212
("Perceptron" , "3"),
1313
("Pooling" , "4"),
1414
("Probabilistic" , "5"),
15-
("LSTM" , "6"),
16-
("Reccurrent" , "7"),
17-
("Unscaling" , "8"),
18-
("Flatten" , "9"),
19-
("Bounding" , "10"),
15+
("BatchNormalization" , "6"),
16+
("LSTM" , "7"),
17+
("Reccurrent" , "8"),
18+
("Unscaling" , "9"),
19+
("Flatten" , "10"),
20+
("Bounding" , "11"),
2021
]
2122
)
2223

0 commit comments

Comments
 (0)