From cc4fbe45b998b866819979add01f5ca64a091cd4 Mon Sep 17 00:00:00 2001 From: Malar Kannan Date: Mon, 11 Dec 2017 15:03:14 +0530 Subject: [PATCH] trying to overfit 2 samples with model -> doesn't seem to converge --- segment_data.py | 2 +- segment_model.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/segment_data.py b/segment_data.py index fb3c28a..521fe02 100644 --- a/segment_data.py +++ b/segment_data.py @@ -258,7 +258,7 @@ if __name__ == '__main__': # plot_segments('story_test_segments') # fix_csv('story_words') # pass - create_segments_tfrecords('story_words', sample_count=0) + create_segments_tfrecords('story_words.3', sample_count=3,train_test_ratio=0.33) # record_generator,input_data,output_data,copy_read_consts = read_segments_tfrecords_generator('story_test') # tr_gen = record_generator() # for i in tr_gen: diff --git a/segment_model.py b/segment_model.py index 8e52f72..d6c8f37 100644 --- a/segment_model.py +++ b/segment_model.py @@ -79,13 +79,13 @@ def load_model_arch(mod_file): def train_segment(collection_name = 'test',resume_weights='',initial_epoch=0): # collection_name = 'story_test' - batch_size = 64 - # batch_size = 4 + # batch_size = 32 + batch_size = 1 model_dir = './models/segment/'+collection_name create_dir(model_dir) log_dir = './logs/segment/'+collection_name create_dir(log_dir) - tr_gen_fn,te_x,te_y,copy_read_consts = read_segments_tfrecords_generator(collection_name,batch_size,2*batch_size) + tr_gen_fn,te_x,te_y,copy_read_consts = read_segments_tfrecords_generator(collection_name,batch_size,batch_size) tr_gen = tr_gen_fn() n_step,n_features,n_records = copy_read_consts(model_dir) input_dim = (n_step, n_features) @@ -137,4 +137,4 @@ def train_segment(collection_name = 'test',resume_weights='',initial_epoch=0): if __name__ == '__main__': # pass - train_segment('story_words')#,'./models/segment/story_phrases.1000/speech_segment_model-final.h5',1001) + train_segment('story_words.3')#,'./models/segment/story_phrases.1000/speech_segment_model-final.h5',1001)