From 240ecb3f271f0bd37d0c4d5ccfe800179eb68ba8 Mon Sep 17 00:00:00 2001 From: Malar Kannan Date: Mon, 11 Dec 2017 14:12:23 +0530 Subject: [PATCH] removed bn output layer --- segment_model.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/segment_model.py b/segment_model.py index e6d5071..4c7a85b 100644 --- a/segment_model.py +++ b/segment_model.py @@ -59,8 +59,7 @@ def simple_segment_model(input_dim): d2 = Dense(8, activation='relu')(bn_d1) bn_d2 = BatchNormalization(momentum=0.98)(d2) d3 = Dense(1, activation='softmax')(bn_d2) - bn_d3 = BatchNormalization(momentum=0.98)(d3) - oup = Reshape(target_shape=(input_dim[0],))(bn_d3) + oup = Reshape(target_shape=(input_dim[0],))(d3) return Model(inp, oup) def write_model_arch(mod,mod_file):