From fea9184aec287796e11d56c33e3171f8ce0521e4 Mon Sep 17 00:00:00 2001 From: Malar Kannan Date: Mon, 11 Dec 2017 13:47:30 +0530 Subject: [PATCH] using the full data and fixed typo in model layer name --- segment_model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/segment_model.py b/segment_model.py index ce0c6cf..cd6caa2 100644 --- a/segment_model.py +++ b/segment_model.py @@ -54,7 +54,7 @@ def simple_segment_model(input_dim): b_gr1 = Bidirectional(GRU(256, return_sequences=True),merge_mode='sum')(inp) # b_gr1 # b_gr2 = Bidirectional(GRU(64, return_sequences=True),merge_mode='sum')(b_gr1) - b_gr3 = Bidirectional(GRU(64, return_sequences=True),merge_mode='sum')(b_gr2) + b_gr3 = Bidirectional(GRU(64, return_sequences=True),merge_mode='sum')(b_gr1) d1 = Dense(32, activation='relu')(b_gr3) d2 = Dense(8, activation='relu')(d1) d3 = Dense(1, activation='softmax')(d2) @@ -131,5 +131,5 @@ def train_segment(collection_name = 'test',resume_weights='',initial_epoch=0): if __name__ == '__main__': - pass - # train_segment('story_phrases','./models/segment/story_phrases.1000/speech_segment_model-final.h5',1001) + # pass + train_segment('story_phrases_full')#,'./models/segment/story_phrases.1000/speech_segment_model-final.h5',1001)