Debuging triplet loss

e22b1822 · Tiago de Freitas Pereira · 80206858 · e22b1822
Commit e22b1822 authored 8 years ago by Tiago de Freitas Pereira
--- a/bob/learn/tensorflow/script/train_mnist_triplet.py
+++ b/bob/learn/tensorflow/script/train_mnist_triplet.py
@@ -132,8 +132,8 @@ def main():
                                 iterations=ITERATIONS,
                                 snapshot=VALIDATION_TEST,
                                 optimizer=optimizer)
-        trainer.train(train_data_shuffler, validation_data_shuffler)
-        #trainer.train(train_data_shuffler)
+        #trainer.train(train_data_shuffler, validation_data_shuffler)
+        trainer.train(train_data_shuffler)
    else:
        mlp = MLP(n_classes, hidden_layers=[15, 20])