Commit 0bf1dd2e authored by Luca Pasa's avatar Luca Pasa
Browse files


parent 965d37a2
......@@ -52,10 +52,11 @@ def serialize_sequence(base_audio_sequence, audio_sequence, video_sequence, labe
def restore_and_get_model(graph, sess, batch_size, nIn_audio, nIn_video, nHidden, nHidden_encode, learningRate,
updating_step, learningDecay, momentum, ckpt_file): # i ckpt_file):
model = DAE_4_speech(graph=graph, n_in_audio=nIn_audio, n_in_video=nIn_video, n_hidden=nHidden,
model = DAE_4_speech(graph=graph, sess=sess, 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,
momentum=momentum, updating_step=updating_step)
saver = tf.train.Saver()
saver.restore(sess, ckpt_file) # "./test_log/init_test-480.ckpt")
return model
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