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

minor update

parent 7b1a4410
......@@ -20,10 +20,10 @@ if __name__ == '__main__':
#Code for running on CPU
config = tf.ConfigProto(device_count={'GPU': 0})
# config = tf.ConfigProto(device_count={'GPU': 0})
graph = tf.Graph()
with tf.Session(graph=graph,config=config) as sess:
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)
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