adding missing model-dir for training constants copying
parent
1f60183ab8
commit
6355db4af7
|
|
@ -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')
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue