adding missing model-dir for training constants copying
This commit is contained in:
@@ -254,7 +254,7 @@ if __name__ == '__main__':
|
|||||||
# create_spectrogram_tfrecords('story_all',sample_count=25)
|
# create_spectrogram_tfrecords('story_all',sample_count=25)
|
||||||
# fix_csv('story_words_test')
|
# fix_csv('story_words_test')
|
||||||
#fix_csv('story_phrases')
|
#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)
|
# create_spectrogram_tfrecords('audio',sample_count=50)
|
||||||
# read_siamese_tfrecords_generator('audio')
|
# read_siamese_tfrecords_generator('audio')
|
||||||
# padd_zeros_siamese_tfrecords('audio')
|
# padd_zeros_siamese_tfrecords('audio')
|
||||||
|
|||||||
@@ -75,7 +75,7 @@ def train_siamese(audio_group = 'audio'):
|
|||||||
log_dir = './logs/'+audio_group
|
log_dir = './logs/'+audio_group
|
||||||
create_dir(log_dir)
|
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)
|
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()
|
tr_gen = tr_gen_fn()
|
||||||
input_dim = (n_step, n_features)
|
input_dim = (n_step, n_features)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user