Commit 18f2ceb5 authored by Luca Pasa's avatar Luca Pasa
Browse files

add test GRID_100

parent 7f4cc5bb
......@@ -6,10 +6,10 @@ import tensorflow as tf
if __name__ == '__main__':
num_epochs = 500
batch_size = 18
batch_size = 30
nIn_audio = 123
nIn_video = 134
nHidden = [350,500]
nHidden = [123,500,600]
nHidden_encode = 750
learningRate = 0.001
traininglog_dir = "./"
......@@ -17,15 +17,15 @@ if __name__ == '__main__':
test_step= 10
learningDecay = 1
momentum = 0.9
test_name="Overfitting_TEST_Concat_DAE_4_speech_Test_lr-"+str(learningRate)+"_batch_size-"+str(batch_size)+"_n_hidden_encode-"+str(nHidden_encode)
test_name="GRID_100_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/multiModalTfRec/TRAIN_CTC_SENTENCES/",
test_set_path="/home/storage/Data/MULTI_GRID/multiModalTfRec/TEST_CTC_SENTENCES/",
validation_set_path="/home/storage/Data/MULTI_GRID/multiModalTfRec/VAL_CTC_SENTENCES/",
model.training_model(training_set_path="/home/storage/Data/MULTI_GRID_100/multiModalTfRec/TRAIN_CTC_SENTENCES/",
test_set_path="/home/storage/Data/MULTI_GRID_100/multiModalTfRec/TEST_CTC_SENTENCES/",
validation_set_path="/home/storage/Data/MULTI_GRID_100/multiModalTfRec/VAL_CTC_SENTENCES/",
n_epoch=num_epochs, test_step=test_step, test_name=test_name,
log_dir="./test_log/")
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