More epochs

indiquo/training/similarity/TrainSimilarity.py

@@ -59,7 +59,7 @@ def main():
     evaluator = evaluation.TripletEvaluator(val_anchor, val_positive, val_negative)
 
     # Tune the model
-    model.fit(train_objectives=[(train_dataloader, train_loss)], epochs=3, warmup_steps=100,
+    model.fit(train_objectives=[(train_dataloader, train_loss)], epochs=5, warmup_steps=200,
               evaluator=evaluator, evaluation_steps=10000, output_path=output_path)