Commit a2eb863d authored by Luca Pasa's avatar Luca Pasa
Browse files

minor mods local

parent 60e0d32b
......@@ -6,7 +6,7 @@ import tensorflow as tf
if __name__ == '__main__':
num_epochs = 500
batch_size = 9
batch_size = 30
nIn_audio = 257
nIn_video = 134
nHidden = [500,600]
......
......@@ -19,11 +19,16 @@ if __name__ == '__main__':
momentum = 0.9
test_name="GRID_100_Concat_DAE_4_speech_Test_lr-"+str(learningRate)+"_batch_size-"+str(batch_size)+"_n_hidden_encode-"+str(nHidden_encode)
#Code for running on CPU
config = tf.ConfigProto(device_count={'GPU': 0})
#------------#
graph = tf.Graph()
with tf.Session(graph=graph) as sess:
with tf.Session(graph=graph,config=config) 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.restore_model()
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/",
......
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