Increase epochs to 10

train_reranker.py

@@ -40,7 +40,7 @@ logger = logging.getLogger(__name__)
 DATASET_NAME = os.environ.get("DATASET_NAME", "amanwithaplan/arcade-reranker-data")
 HUB_MODEL_ID = os.environ.get("HUB_MODEL_ID", "amanwithaplan/arcade-reranker")
 BASE_MODEL = os.environ.get("BASE_MODEL", "Alibaba-NLP/gte-reranker-modernbert-base")
-NUM_EPOCHS = int(os.environ.get("NUM_EPOCHS", "5"))
+NUM_EPOCHS = int(os.environ.get("NUM_EPOCHS", "10"))
 BATCH_SIZE = int(os.environ.get("BATCH_SIZE", "16"))
 LEARNING_RATE = float(os.environ.get("LEARNING_RATE", "2e-5"))
 MAX_SEQ_LENGTH = int(os.environ.get("MAX_SEQ_LENGTH", "512"))