[test_lstm] Add logging

bob/learn/tensorflow/test/test_dnn.py

@@ -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

bob/learn/tensorflow/test/test_lstm.py

@@ -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")