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

Debuging triplet loss

parent 80206858
...@@ -132,8 +132,8 @@ def main(): ...@@ -132,8 +132,8 @@ def main():
iterations=ITERATIONS, iterations=ITERATIONS,
optimizer=optimizer) optimizer=optimizer)
trainer.train(train_data_shuffler, validation_data_shuffler) #trainer.train(train_data_shuffler, validation_data_shuffler)
#trainer.train(train_data_shuffler) trainer.train(train_data_shuffler)
else: else:
mlp = MLP(n_classes, hidden_layers=[15, 20]) mlp = MLP(n_classes, hidden_layers=[15, 20])
Supports Markdown
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