From 9132c24b560e8d1ea970bd361acc77c08235c225 Mon Sep 17 00:00:00 2001
From: Frederik Arnold <frederik.arnold@hu-berlin.de>
Date: Wed, 24 Jan 2024 09:00:19 +0100
Subject: [PATCH] More epochs
---
indiquo/training/similarity/TrainSimilarity.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/indiquo/training/similarity/TrainSimilarity.py b/indiquo/training/similarity/TrainSimilarity.py
index d7981bd..d71cec5 100644
--- a/indiquo/training/similarity/TrainSimilarity.py
+++ b/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)
--
GitLab