Commit 918c0d9d authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Updated test case

parent f6a946f0
Pipeline #12479 passed with stages
in 18 minutes and 8 seconds
......@@ -102,7 +102,7 @@ def test_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, trainer.graph)
accuracy = validate_network(embedding, validation_data, validation_labels)
......
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