Commit 9ff37986 authored by Tiago Pereira's avatar Tiago Pereira

Triplet networks trainer with prefetch #27

parent 144ce19b
Pipeline #11265 passed with stages
in 18 minutes and 23 seconds
......@@ -178,7 +178,9 @@ def test_tripletcnn_trainer():
optimizer=tf.train.GradientDescentOptimizer(0.01),)
trainer.train(train_data_shuffler)
embedding = Embedding(train_data_shuffler("data", from_queue=False)['anchor'], graph['anchor'])
embedding = Embedding(validation_data_shuffler("data", from_queue=False),
architecture(validation_data_shuffler("data", from_queue=False), reuse=True))
eer = dummy_experiment(validation_data_shuffler, embedding)
assert eer < 0.15
shutil.rmtree(directory)
......
Markdown is supported
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