adding missing model-dir for training constants copying

master
Malar Kannan 2017-11-22 15:04:02 +05:30
parent 1f60183ab8
commit 6355db4af7
2 changed files with 2 additions and 2 deletions

View File

@ -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')

View File

@ -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)