Commit 53053a10 authored by Luca Pasa's avatar Luca Pasa
Browse files

bug fix spectogram test

parent 6cc3e87d
......@@ -17,15 +17,15 @@ if __name__ == '__main__':
test_step= 10
learningDecay = 1
momentum = 0.9
test_name="GRID_100_Spectro_Concat_DAE_4_speech_Test_lr-"+str(learningRate)+"_batch_size-"+str(batch_size)+"_n_hidden_encode-"+str(nHidden_encode)
test_name="MULTI_GRID_100_Spectro_Concat_DAE_4_speech_Test_lr-"+str(learningRate)+"_batch_size-"+str(batch_size)+"_n_hidden_encode-"+str(nHidden_encode)
graph = tf.Graph()
with tf.Session(graph=graph) as sess:
model = DAE_4_speech(sess=sess, graph=graph, n_in_audio=nIn_audio, n_in_video=nIn_video, n_hidden=nHidden,
n_hidden_encode=nHidden_encode, batch_size=batch_size, learning_rate=learningRate,
learning_decay=learningDecay, momentum=momentum, updating_step=updating_step)
model.training_model(training_set_path="/home/storage/Data/MULTI_GRID/rawMultiModalTfRec/TRAIN_CTC_SENTENCES/",
test_set_path="/home/storage/Data/MULTI_GRID/rawMultiModalTfRec/TEST_CTC_SENTENCES/",
validation_set_path="/home/storage/Data/MULTI_GRID/rawMultiModalTfRec/VAL_CTC_SENTENCES/",
model.training_model(training_set_path="/home/storage/Data/MULTI_GRID_100/rawMultiModalTfRec/TRAIN_CTC_SENTENCES/",
test_set_path="/home/storage/Data/MULTI_GRID_100/rawMultiModalTfRec/TEST_CTC_SENTENCES/",
validation_set_path="/home/storage/Data/MULTI_GRID_100/rawMultiModalTfRec/VAL_CTC_SENTENCES/",
n_epoch=num_epochs, test_step=test_step, test_name=test_name,
log_dir="./test_log/")
......@@ -102,7 +102,7 @@ class Bi_DEA_model:
# fw_cell:
cells_fw = []
for dim in reversed(self.n_hidden):
cells_fw.append(tf.contrib.rnn.BasicLSTMCell(dim))
cells_fw.append(tf.contrib.rnn.BasicLSTMCell(dim))
#cells_fw.append(tf.contrib.rnn.DropoutWrapper(tf.contrib.rnn.BasicLSTMCell(dim),
#output_keep_prob=self.keep_prob, input_keep_prob=self.keep_prob))
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment