Skip to content
Snippets Groups Projects
Commit 22df299b authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Fixes some unit tests

parent e6864f8a
No related branches found
No related tags found
1 merge request!14gridtk integration
Pipeline #
......@@ -173,7 +173,7 @@ def test_triplet_cnn_pretrained():
analizer=None,
temp_dir=directory)
trainer.create_network_from_file(os.path.join(directory, "model.ckp"))
trainer.create_network_from_file(os.path.join(directory, "model.ckp.meta"))
trainer.train()
embedding = Embedding(trainer.data_ph['anchor'], trainer.graph['anchor'])
......@@ -189,6 +189,7 @@ def test_triplet_cnn_pretrained():
def test_siamese_cnn_pretrained():
tf.reset_default_graph()
train_data, train_labels, validation_data, validation_labels = load_mnist()
......@@ -244,7 +245,7 @@ def test_siamese_cnn_pretrained():
analizer=None,
temp_dir=directory)
trainer.create_network_from_file(os.path.join(directory, "model.ckp"))
trainer.create_network_from_file(os.path.join(directory, "model.ckp.meta"))
trainer.train()
#embedding = Embedding(train_data_shuffler("data", from_queue=False)['left'], trainer.graph['left'])
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment