Skip to content
Snippets Groups Projects
Commit 1fb81237 authored by Olivier Canévet's avatar Olivier Canévet
Browse files

[test_lstm] Add logging

parent 92f3cb71
No related branches found
No related tags found
No related merge requests found
......@@ -32,7 +32,6 @@ def validate_network(embedding, validation_data, validation_labels):
[data, labels] = validation_data_shuffler.get_batch()
predictions = embedding(data)
accuracy = 100. * numpy.sum(numpy.argmax(predictions, axis=1) == labels) / predictions.shape[0]
return accuracy
......@@ -75,7 +74,7 @@ def test_dnn_trainer():
# At least 50% of accuracy for the DNN
assert accuracy > 50.
shutil.rmtree(directory)
#shutil.rmtree(directory)
del architecture
del trainer # Just to clean the variables
......@@ -8,4 +8,10 @@ from bob.learn.tensorflow.trainers import Trainer, constant
from bob.learn.tensorflow.utils import load_mnist
import tensorflow as tf
import bob.core
logger = bob.core.log.setup("LSTM") # bob.learn.tensorflow does not work
bob.core.log.set_verbosity_level(logger, 3)
# Data
logger.debug("Loading MNIST")
train_data, train_labels, validation_data, validation_labels = load_mnist(data_dir="mnist")
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment