diff --git a/speech_data.py b/speech_data.py
index cc91b7e..cc171ce 100644
--- a/speech_data.py
+++ b/speech_data.py
@@ -254,7 +254,7 @@ if __name__ == '__main__':
     # create_spectrogram_tfrecords('story_all',sample_count=25)
     # fix_csv('story_words_test')
     #fix_csv('story_phrases')
-    create_spectrogram_tfrecords('story_phrases',sample_count=100,train_test_ratio=0.1)
+    create_spectrogram_tfrecords('story_phrases',sample_count=500,train_test_ratio=0.1)
     # create_spectrogram_tfrecords('audio',sample_count=50)
     # read_siamese_tfrecords_generator('audio')
     # padd_zeros_siamese_tfrecords('audio')
diff --git a/speech_model.py b/speech_model.py
index 42a6fb3..ce2a7c0 100644
--- a/speech_model.py
+++ b/speech_model.py
@@ -75,7 +75,7 @@ def train_siamese(audio_group = 'audio'):
     log_dir = './logs/'+audio_group
     create_dir(log_dir)
     tr_gen_fn,te_pairs,te_y,copy_read_consts = read_siamese_tfrecords_generator(audio_group,batch_size=batch_size,test_size=batch_size)
-    n_step,n_features,n_records = copy_read_consts()
+    n_step,n_features,n_records = copy_read_consts(model_dir)
     tr_gen = tr_gen_fn()
     input_dim = (n_step, n_features)