Commit 7b1a4410 authored by Luca Pasa's avatar Luca Pasa
Browse files

update grid100 test

parent 9feee5ef
......@@ -5,7 +5,7 @@ from Concat_DAE_4_speech import DAE_4_speech
import tensorflow as tf
if __name__ == '__main__':
num_epochs = 500
num_epochs = 120
batch_size = 15
nIn_audio = 123
nIn_video = 134
......@@ -28,7 +28,7 @@ if __name__ == '__main__':
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.restore_model("./RESULT/BaseLine1/GRID_100_Concat_DAE_4_speech_Test_lr-0.001_batch_size-15_n_hidden_encode-750.ckpt-80")
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/",
......
......@@ -143,7 +143,7 @@ class DAE_4_speech:
except tf.errors.OutOfRangeError:
print("End of test")
print("End of test, PC: "+np.mean(np.asarray(pearson_cor_list)))
break
return np.mean(np.asarray(pearson_cor_list)) , np.mean(np.asarray(ecl_dist_list))
......
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