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
Branches
Tags
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