modernbert-embed-base-dj-ft-v2 / README.md

dujun

Add new SentenceTransformer model

8f77d52 verified 9 months ago

preview code

raw

history blame contribute delete

40.4 kB

metadata

tags:
  - sentence-transformers
  - sentence-similarity
  - feature-extraction
  - generated_from_trainer
  - dataset_size:274
  - loss:MultipleNegativesRankingLoss
base_model: nomic-ai/modernbert-embed-base
widget:
  - source_sentence: >-
      Besides penalties, what other consequences might result from continuous
      rule breaches?
    sentences:
      - "20  Misconduct  \n20.1\tMisconduct warranting Penalty, Forced Interchange, Sin Bin or Dismissal \nincludes:\n20.1.1\tContinuous or regular breaches of the Rules;\n20.1.2\tSwearing towards another player, Referee, spectator or other match \t\nofficial;\n20.1.3\tDisputing decisions of Referees or other match official(s);\n20.1.4\tUsing more than the necessary physical force to make a Touch;\n20.1.5\tPoor sportsmanship;\n20.1.6\tTripping, striking, or otherwise assaulting another player, Referee, \nspectator or other match official; or\n20.1.7\tAny other action that is contrary to the spirit of the game.21  Forced Interchange  \n21.1\tWhere the Referee deems it necessary to implement a Forced Interchange \nfollowing an Infringement, the Referee is to stop the match, direct the ball to \nbe placed on the Mark, advise the offending player of the reason for the Forced \nInterchange, direct that player to return to the Interchange Area, display the \nrelevant signal and award a Penalty to the non-offending Team."
      - "Line \nMarkings are to be laid out as shown in Appendix 1 - The Field of Play.Sidelines \nextend seven (7) metres beyond the Try Lines to join the Dead Ball Lines and \ndefine the In-Goal Areas which measure fifty (50) metres wide by seven (7) \nmetres in length.1.3\tThe Interchange Areas are located no closer than one (1) metre from each \nSideline.1.4\tSuitably sized markers, cones or corner posts of a distinguishing colour and \nmade from safe and pliable material should be positioned at the intersections of \nthe Sideline and Halfway line and the Sideline and the Try Line.1.4.1\tMarkers, cones or corner posts placed on the junction of the Sideline and \nTry Line are deemed to be in the Field of Play.1.4.2\tAll other markers or cones are deemed to be out of the Field of Play.1.5\tThe standard playing surface is grass.Other surfaces including synthetic grass \nmay be used but shall be subject to NTA approved standards."
      - "The ball may be passed, knocked or handed between players \nof the Attacking Team who may in turn run or otherwise move with the ball in an \nattempt to gain territorial Advantage and to score Tries.Defending players prevent \nthe Attacking Team from gaining a territorial Advantage by touching the ball carrier.1  The Field of Play  \n  \n1.1\tThe Field of Play is rectangular in shape measuring 70 metres in length from \nTry Line to Try Line, excluding the In-Goal Areas and 50 metres in width from \nSideline to Sideline excluding the Interchange Areas.1.1.1\tVariations to the dimensions of the Field of Play may be made but must be \nincluded in relevant competition, event or tournament conditions\n1.2\tLine Markings should be 4cm in width but must be no less than 2.5cm.Line \nMarkings are to be laid out as shown in Appendix 1 - The Field of Play."
  - source_sentence: Does the player receiving the ball in a Rollball become the Half?
    sentences:
      - "Ruling = A Penalty to the Defending Team at the point of the Infringement.13.5\tA player may only perform a Rollball at the Mark under the following \ncircumstances:\n13.5.1\twhen a Touch has been made; or\n13.5.2\twhen Possession changes following the sixth Touch; or\n13.5.3\twhen Possession changes due to the ball being dropped or passed and \ngoes to the ground; or\n13.5.4\twhen Possession changes due to an Infringement by an attacking player \nat a Penalty, a Tap or a Rollball; or\nFIT Playing Rules - 5th Edition\nCOPYRIGHT © Touch Football Australia 2020\n11\n13.5.5\twhen Possession changes after the Half is Touched or when the Half \nplaces the ball on or over the Try Line; or\n13.5.6\tin replacement of a Penalty Tap; or\n13.5.7\twhen so directed by the Referee."
      - "4.4\tHats or caps are permitted to be worn during a match provided they are safe \nand meet any NTA regulations.4.5\tSafe footwear must be worn with exceptions allowed for game variants such as \nBeach Touch.4.6\tLight leather or synthetic boots with soft moulded soles are permitted.4.6.1\tShoes with screw-in studs are not to be worn by any player or Referee.4.7\tPlayers are not to participate in any match wearing any item of jewellery, \nchain, identification band/bracelet or similar item that may prove dangerous.Any jewellery or other items that cannot be removed are to be taped to the \nsatisfaction of the Referee.4.8\tLong (extend beyond the finger flesh when viewed from the palm) or sharp \nfingernails are not allowed.4.9\tReferees and players may wear spectacles or sunglasses provided they are safe \nand securely attached."
      - "FIT Playing Rules - 5th Edition\nCOPYRIGHT © Touch Football Australia 2020\n15\n19  Advantage  \n19.1\tWhere a Defending Team player is Offside at a Tap or Rollball and attempts \nto interfere with play, the Referee will allow Advantage or award a Penalty, \nwhichever is of greater Advantage to the Attacking Team.19.2\tShould the Attacking Team in the act of taking the Advantage subsequently \ninfringe, the Ruling on the initial Infringement will apply.20  Misconduct  \n20.1\tMisconduct warranting Penalty, Forced Interchange, Sin Bin or Dismissal \nincludes:\n20.1.1\tContinuous or regular breaches of the Rules;\n20.1.2\tSwearing towards another player, Referee, spectator or other match \t\nofficial;\n20.1.3\tDisputing decisions of Referees or other match official(s);\n20.1.4\tUsing more than the necessary physical force to make a Touch;\n20.1.5\tPoor sportsmanship;\n20.1.6\tTripping, striking, or otherwise assaulting another player, Referee, \nspectator or other match official; or\n20.1.7\tAny other action that is contrary to the spirit of the game."
  - source_sentence: Is there a limit to how many times a player can interchange?
    sentences:
      - >-
        Except as permitted under the 

        Copyright Act, these Rules must not be reproduced by any process,
        electronic or otherwise, without the written 

        permission of Touch Football Australia.Attacking Try Line

        The line on or over which a player has to place the ball to 

        score a Try.Attacking Team

        The Team which has or is gaining Possession.Behind

        A position or direction towards a Team’s Defending Try Line.Change of
        Possession

        The act of moving control of the ball from one Team to the
        other.Dead/Dead Ball

        When the ball is out of play including the period following a Try and 

        until the match is recommenced and when the ball goes to ground 

        and/or outside the boundaries of the Field of Play prior to the 

        subsequent Rollball.Dead Ball Line

        The end boundaries of the Field of Play.There is one at each end of 

        the Field of Play.See Appendix 1.Defending Try Line

        The line which a Team has to defend to prevent a Try.Defending Team

        The Team without or which is losing Possession.
      - "12.5\tIf a player from the Defending Team unintentionally makes contact with the ball \nin flight and the ball goes to ground, a Change of Possession occurs.12.6\tIf a player from the Defending Team unintentionally makes contact with the \nball in flight and the ball is retrieved by an attacking player, play and the Touch \nCount continues.12.7\tA player from the Attacking Team cannot pass the ball into a defending player \nintentionally seeking a rebound or a restart of the Touch count.Ruling = A Penalty to the Defending Team at the point of the pass.13  The Rollball  \n  \n13.1\tThe attacking player is to position on the Mark, face the opponent’s Try Line, \nmake a genuine attempt to stand parallel to the Sidelines, place the ball on the \nground between the feet in a controlled manner and:\n13.1.1\tstep Forward over the ball; or\n13.1.2\troll the ball back between the feet no more than one (1) metre; or\n13.1.3\tpass a foot over the ball."
      - "Ruling = A Penalty to the non-offending Team at the point of the Infringement.16.3\tAn attacking player in support of the player in Possession may move as \nnecessary to achieve a supporting position but must not grab, hold, push or \notherwise  deliberately interfere with a defending player attempting to make a \nTouch\nRuling = A Penalty to the non-offending Team at the point of the Infringement or on the  \nseven (7) metre line if the Infringement occurs in the In-Goal Area.16.4\tPlayers in the Defending Team may not obstruct or interfere with an attacking \nplayer.Ruling = A Penalty to the non-offending Team at the point of the Infringement or on the \nseven (7) metre line if the Infringement occurs in the In-Goal Area.16.5\tShould a supporting, attacking player cause an apparent and involuntary or \naccidental Obstruction and the player in Possession ceases movement to allow \na Touch to be made, the Touch is to count."
  - source_sentence: >-
      Where is a penalty awarded to the attacking team if a defending player
      interferes with them after a touch?
    sentences:
      - "Ruling = A Penalty to the non-offending Team at the point of the Infringement.16.3\tAn attacking player in support of the player in Possession may move as \nnecessary to achieve a supporting position but must not grab, hold, push or \notherwise  deliberately interfere with a defending player attempting to make a \nTouch\nRuling = A Penalty to the non-offending Team at the point of the Infringement or on the  \nseven (7) metre line if the Infringement occurs in the In-Goal Area.16.4\tPlayers in the Defending Team may not obstruct or interfere with an attacking \nplayer.Ruling = A Penalty to the non-offending Team at the point of the Infringement or on the \nseven (7) metre line if the Infringement occurs in the In-Goal Area.16.5\tShould a supporting, attacking player cause an apparent and involuntary or \naccidental Obstruction and the player in Possession ceases movement to allow \na Touch to be made, the Touch is to count."
      - "16.5\tShould a supporting, attacking player cause an apparent and involuntary or \naccidental Obstruction and the player in Possession ceases movement to allow \na Touch to be made, the Touch is to count.16.6\tIf the Referee causes Obstruction on either an attacking player or a defending \nplayer including when the ball makes contact with the Referee, play should \npause and recommence with a Rollball at the Mark where the interference \noccurred and the Touch count remains unchanged.17  Interchange  \n17.1\tPlayers may Interchange at any time.17.2\tThere is no limit to the number of times a player may Interchange.17.3\tInterchange players must remain in their Interchange Area for the duration of \nthe match.17.4\tInterchanges may only occur after the player leaving the Field of Play has \nentered the Interchange Area.17.5\tPlayers leaving or entering the Field of Play shall not hinder or obstruct play.Ruling = A Penalty to the non-offending Team at the point of the Infringement."
      - "11.2\tA player in Possession  may not intentionally kick, pass, flick, knock, throw, \nhand-off or otherwise propel the ball in a Forward direction over an opposition \nplayer and regain Possession.Ruling = A Penalty will be awarded to the Defending Team at the Mark where the ball was \npropelled Forward.12  Ball Touched in Flight  \n  \n12.1\tIf a player from the Defending Team deliberately makes contact with the ball in \nflight and the ball goes to ground, the Attacking Team retains the ball and the \nTouch Count restarts as zero (0) Touch.12.2\tIf a player from the Defending Team deliberately makes contact with the ball \nin flight and the ball is retrieved by an attacking player, without touching the \nground, play continues and the next Touch is zero (0) Touch."
  - source_sentence: When does a player cease to be the Half?
    sentences:
      - "Ruling = A Penalty to the non-offending Team at the point of the Infringement.17.6\tPlayers entering the Field of Play must take up an Onside position before \nbecoming involved in play.FIT Playing Rules - 5th Edition\n14\nCOPYRIGHT © Touch Football Australia 2020\nRuling = A Penalty to the non-offending Team at the point of the Infringement.17.7\tWhen an intercept has occurred or a line break made, players are not permitted \nto Interchange until the next Touch has been made or ball becomes Dead.Ruling A = If a player enters the Field of Play and prevents the scoring of a Try, a Penalty Try \nwill be awarded and the offending player sent to the Sin Bin.Ruling B = If a player enters the Field of Play but does not impede the scoring of a Try the \noffending player will be sent to the Sin Bin.17.8\tFollowing a Try, players may Interchange at will, without having to wait for the \nplayer to enter the Interchange Area, but must do so prior to the Tap being taken \nto recommence play."
      - "18.7\tA player may perform a Rollball instead of a Penalty Tap and the player who \nreceives the ball does not become the Half.18.8\tIf the Defending Team is penalised three (3) times upon entering their Seven \nMetre Zone during a single Possession, the last offending player will be given an \nExclusion until the end of that Possession.18.9\tA Penalty Try is awarded if any action by a player, Team official or spectator, \ndeemed by the Referee to be contrary to the Rules or spirit of the game clearly \nprevents the Attacking Team from scoring a Try.FIT Playing Rules - 5th Edition\nCOPYRIGHT © Touch Football Australia 2020\n15\n19  Advantage  \n19.1\tWhere a Defending Team player is Offside at a Tap or Rollball and attempts \nto interfere with play, the Referee will allow Advantage or award a Penalty, \nwhichever is of greater Advantage to the Attacking Team."
      - "13.10\tA player ceases to be the Half once the ball is passed to another player.13.11\tDefending players are not to interfere with the performance of the Rollball or the \nHalf.Ruling = A Penalty to the Attacking Team at a point ten (10) metres directly Forward of the \nInfringement.13.12\tPlayers of the Defending Team must not move Forward of the Onside position \nuntil the Half has made contact with the ball, unless directed to do so by the \nReferee or in accordance with 13.12.1.13.12.1\tWhen the Half is not within one (1) metre of the Rollball, Onside players \nof the Defending Team may move Forward as soon as the player \nperforming the Rollball releases the ball.If the Half is not in position and \na defending player moves Forward and makes contact with the ball, a \nChange of Possession results."
datasets:
  - Trelis/touch-rugby-modernbert-pairs
pipeline_tag: sentence-similarity
library_name: sentence-transformers

SentenceTransformer based on nomic-ai/modernbert-embed-base

This is a sentence-transformers model finetuned from nomic-ai/modernbert-embed-base on the touch-rugby-modernbert-pairs dataset. It maps sentences & paragraphs to a 768-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.

Model Details

Model Description

Model Type: Sentence Transformer
Base model: nomic-ai/modernbert-embed-base
Maximum Sequence Length: 8192 tokens
Output Dimensionality: 768 dimensions
Similarity Function: Cosine Similarity
Training Dataset:
- touch-rugby-modernbert-pairs

Model Sources

Documentation: Sentence Transformers Documentation
Repository: Sentence Transformers on GitHub
Hugging Face: Sentence Transformers on Hugging Face

Full Model Architecture

SentenceTransformer(
  (0): Transformer({'max_seq_length': 8192, 'do_lower_case': False}) with Transformer model: ModernBertModel 
  (1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
  (2): Normalize()
)

Usage

Direct Usage (Sentence Transformers)

First install the Sentence Transformers library:

pip install -U sentence-transformers

Then you can load this model and run inference.

from sentence_transformers import SentenceTransformer

# Download from the 🤗 Hub
model = SentenceTransformer("dujun/modernbert-embed-base-dj-ft-v2")
# Run inference
sentences = [
    'When does a player cease to be the Half?',
    '13.10\tA player ceases to be the Half once the ball is passed to another player.13.11\tDefending players are not to interfere with the performance of the Rollball or the \nHalf.Ruling = A Penalty to the Attacking Team at a point ten (10) metres directly Forward of the \nInfringement.13.12\tPlayers of the Defending Team must not move Forward of the Onside position \nuntil the Half has made contact with the ball, unless directed to do so by the \nReferee or in accordance with 13.12.1.13.12.1\tWhen the Half is not within one (1) metre of the Rollball, Onside players \nof the Defending Team may move Forward as soon as the player \nperforming the Rollball releases the ball.If the Half is not in position and \na defending player moves Forward and makes contact with the ball, a \nChange of Possession results.',
    '18.7\tA player may perform a Rollball instead of a Penalty Tap and the player who \nreceives the ball does not become the Half.18.8\tIf the Defending Team is penalised three (3) times upon entering their Seven \nMetre Zone during a single Possession, the last offending player will be given an \nExclusion until the end of that Possession.18.9\tA Penalty Try is awarded if any action by a player, Team official or spectator, \ndeemed by the Referee to be contrary to the Rules or spirit of the game clearly \nprevents the Attacking Team from scoring a Try.FIT Playing Rules - 5th Edition\nCOPYRIGHT © Touch Football Australia 2020\n15\n19\u2002 Advantage  \n19.1\tWhere a Defending Team player is Offside at a Tap or Rollball and attempts \nto interfere with play, the Referee will allow Advantage or award a Penalty, \nwhichever is of greater Advantage to the Attacking Team.',
]
embeddings = model.encode(sentences)
print(embeddings.shape)
# [3, 768]

# Get the similarity scores for the embeddings
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [3, 3]

Training Details

Training Dataset

touch-rugby-modernbert-pairs

Dataset: touch-rugby-modernbert-pairs at 7cb0ae2
Size: 274 training samples
Columns: question and related_chunk
Approximate statistics based on the first 274 samples:
question related_chunk
type string string
details
min: 10 tokens
mean: 18.51 tokens
max: 36 tokens

min: 147 tokens
mean: 230.74 tokens
max: 319 tokens

	question	related_chunk
type	string	string
details	min: 10 tokens mean: 18.51 tokens max: 36 tokens	min: 147 tokens mean: 230.74 tokens max: 319 tokens

Samples:

question	related_chunk
`Where does a change of possession occur if a touch is made in In-Goal?`	Ruling = A Penalty to the non-offending Team at the point of the Infringement.10.4 If the ball is accidentally knocked from the hands of a player in Possession during a Touch, the Touch counts and the Attacking Team retains Possession.10.5 The defending player must not deliberately knock the ball from the hands of a player in Possession during a Touch.Ruling = A Penalty to the Attacking Team at the point of the Infringement.10.6 A player must not pass or otherwise deliver the ball after a Touch has been made.Ruling = A Penalty to the Defending Team at the point of the Infringement, or if In-Goal the nearest point on the seven (7) metre line.10.7 The Half may pass or run with the ball but cannot get Touched while in Possession of the ball.Ruling = A Change of Possession occurs at the point of the Touch, or if In-Goal the nearest point on the seven (7) metre line.
`What section details the field of play in the Touch Rugby Rules 5th Edition?`	FIT Playing Rules - 5th Edition COPYRIGHT © Touch Football Australia 2020 Appendix 1 – Field of Play Contents 01 I The Field of Play 5 02 I Player Registration 5 03 I The Ball 6 04 I Playing Uniform 6 05 I Team Composition 6 06 I Team Coach and Team Officials 7 07 I Commencement and Recommencement of Play 7 08 I Match Duration 8 09 I Possession 8 10 I The Touch 9 11 I Passing 10 12 I Ball Touched in Flight 10 13 I The Rollball 11 14 I Scoring 13 15 I Offside 13 16 I Obstruction 14 17 I Interchange 14 18 I Penalty 15 19 I Advantage 16 20 I Misconduct 16 21 I Forced Interchange 16 22 I Sin Bin 16 23 I Dismissal 17
`What is one of the Referee's responsibilities before the match commences?`	`An approach may only be made during a break in play or at the discretion of the Referee.FIT Playing Rules - 5th Edition 18 COPYRIGHT © Touch Football Australia 2020 HALFWAY LINE SIN BIN AREAS IN-GOAL AREA TRY LINE 7 M ZONE DEAD BALL LINE PERIMETER INTERCHANGE AREA 20M 10M 10M 1M 5M 7 M 7 M 7 M 7 M 50M 3M 70M INTERCHANGE AREA Appendix 1 – Field of Play FIT Playing Rules - 5th Edition COPYRIGHT © Touch Football Australia 2020 19 FEDERATION OF INTERNATIONAL TOUCH`

Loss: MultipleNegativesRankingLoss with these parameters:

{
    "scale": 20.0,
    "similarity_fct": "cos_sim"
}

Evaluation Dataset

touch-rugby-modernbert-pairs

Dataset: touch-rugby-modernbert-pairs at 7cb0ae2
Size: 31 evaluation samples
Columns: question and related_chunk
Approximate statistics based on the first 31 samples:
question related_chunk
type string string
details
min: 12 tokens
mean: 19.52 tokens
max: 33 tokens

min: 170 tokens
mean: 234.45 tokens
max: 304 tokens

	question	related_chunk
type	string	string
details	min: 12 tokens mean: 19.52 tokens max: 33 tokens	min: 170 tokens mean: 234.45 tokens max: 304 tokens

Samples:

question	related_chunk
`Where must a player's identifying number be displayed?`	3.2 The ball shall be inflated to the manufacturers’ recommended air pressure.3.3 The Referee shall immediately pause the match if the size and shape of the ball no longer complies with clauses 3.1 or 3.2 to allow for the ball to replaced or the issue rectified.3.4 The ball must not be hidden under player attire.4 Playing Uniform 4.1 Participating players are to be correctly attired in matching Team uniforms 4.2 Playing uniforms consist of shirt, singlet or other item as approved by the NTA or NTA competition provider, shorts and/or tights and socks.4.3 All players are to wear a unique identifying number not less than 16cm in height, clearly displayed on the rear of the playing top.4.3.1 Identifying numbers must feature no more than two (2) digits.4.4 Hats or caps are permitted to be worn during a match provided they are safe and meet any NTA regulations.
`Besides penalties, what other consequences might result from continuous rule breaches?`	20 Misconduct 20.1 Misconduct warranting Penalty, Forced Interchange, Sin Bin or Dismissal includes: 20.1.1 Continuous or regular breaches of the Rules; 20.1.2 Swearing towards another player, Referee, spectator or other match official; 20.1.3 Disputing decisions of Referees or other match official(s); 20.1.4 Using more than the necessary physical force to make a Touch; 20.1.5 Poor sportsmanship; 20.1.6 Tripping, striking, or otherwise assaulting another player, Referee, spectator or other match official; or 20.1.7 Any other action that is contrary to the spirit of the game.21 Forced Interchange 21.1 Where the Referee deems it necessary to implement a Forced Interchange following an Infringement, the Referee is to stop the match, direct the ball to be placed on the Mark, advise the offending player of the reason for the Forced Interchange, direct that player to return to the Interchange Area, display the relevant signal and award a Penalty to the non-offending Team.
`Can a Rollball be performed after a Touch has been made?`	Ruling = A Penalty to the Defending Team at the point of the Infringement.13.5 A player may only perform a Rollball at the Mark under the following circumstances: 13.5.1 when a Touch has been made; or 13.5.2 when Possession changes following the sixth Touch; or 13.5.3 when Possession changes due to the ball being dropped or passed and goes to the ground; or 13.5.4 when Possession changes due to an Infringement by an attacking player at a Penalty, a Tap or a Rollball; or FIT Playing Rules - 5th Edition COPYRIGHT © Touch Football Australia 2020 11 13.5.5 when Possession changes after the Half is Touched or when the Half places the ball on or over the Try Line; or 13.5.6 in replacement of a Penalty Tap; or 13.5.7 when so directed by the Referee.

Loss: MultipleNegativesRankingLoss with these parameters:

{
    "scale": 20.0,
    "similarity_fct": "cos_sim"
}

Training Hyperparameters

Non-Default Hyperparameters

eval_strategy: steps
per_device_train_batch_size: 32
per_device_eval_batch_size: 32
learning_rate: 5e-06
num_train_epochs: 1
lr_scheduler_type: constant
warmup_ratio: 0.3

All Hyperparameters

Click to expand

overwrite_output_dir: False
do_predict: False
eval_strategy: steps
prediction_loss_only: True
per_device_train_batch_size: 32
per_device_eval_batch_size: 32
per_gpu_train_batch_size: None
per_gpu_eval_batch_size: None
gradient_accumulation_steps: 1
eval_accumulation_steps: None
torch_empty_cache_steps: None
learning_rate: 5e-06
weight_decay: 0.0
adam_beta1: 0.9
adam_beta2: 0.999
adam_epsilon: 1e-08
max_grad_norm: 1.0
num_train_epochs: 1
max_steps: -1
lr_scheduler_type: constant
lr_scheduler_kwargs: {}
warmup_ratio: 0.3
warmup_steps: 0
log_level: passive
log_level_replica: warning
log_on_each_node: True
logging_nan_inf_filter: True
save_safetensors: True
save_on_each_node: False
save_only_model: False
restore_callback_states_from_checkpoint: False
no_cuda: False
use_cpu: False
use_mps_device: False
seed: 42
data_seed: None
jit_mode_eval: False
use_ipex: False
bf16: False
fp16: False
fp16_opt_level: O1
half_precision_backend: auto
bf16_full_eval: False
fp16_full_eval: False
tf32: None
local_rank: 0
ddp_backend: None
tpu_num_cores: None
tpu_metrics_debug: False
debug: []
dataloader_drop_last: False
dataloader_num_workers: 0
dataloader_prefetch_factor: None
past_index: -1
disable_tqdm: False
remove_unused_columns: True
label_names: None
load_best_model_at_end: False
ignore_data_skip: False
fsdp: []
fsdp_min_num_params: 0
fsdp_config: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
tp_size: 0
fsdp_transformer_layer_cls_to_wrap: None
accelerator_config: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
deepspeed: None
label_smoothing_factor: 0.0
optim: adamw_torch
optim_args: None
adafactor: False
group_by_length: False
length_column_name: length
ddp_find_unused_parameters: None
ddp_bucket_cap_mb: None
ddp_broadcast_buffers: False
dataloader_pin_memory: True
dataloader_persistent_workers: False
skip_memory_metrics: True
use_legacy_prediction_loop: False
push_to_hub: False
resume_from_checkpoint: None
hub_model_id: None
hub_strategy: every_save
hub_private_repo: None
hub_always_push: False
gradient_checkpointing: False
gradient_checkpointing_kwargs: None
include_inputs_for_metrics: False
include_for_metrics: []
eval_do_concat_batches: True
fp16_backend: auto
push_to_hub_model_id: None
push_to_hub_organization: None
mp_parameters:
auto_find_batch_size: False
full_determinism: False
torchdynamo: None
ray_scope: last
ddp_timeout: 1800
torch_compile: False
torch_compile_backend: None
torch_compile_mode: None
include_tokens_per_second: False
include_num_input_tokens_seen: False
neftune_noise_alpha: None
optim_target_modules: None
batch_eval_metrics: False
eval_on_start: False
use_liger_kernel: False
eval_use_gather_object: False
average_tokens_across_devices: False
prompts: None
batch_sampler: batch_sampler
multi_dataset_batch_sampler: proportional

Training Logs

Epoch	Step	Training Loss	Validation Loss
0.2	1	-	2.7507
0.4	2	3.6185	2.7254
0.6	3	-	2.7059
0.8	4	3.4585	2.6828
1.0	5	-	2.6653

Framework Versions

Python: 3.11.11
Sentence Transformers: 4.1.0
Transformers: 4.51.3
PyTorch: 2.7.0+cu126
Accelerate: 1.7.0
Datasets: 2.17.1
Tokenizers: 0.21.1

Citation

BibTeX

Sentence Transformers

@inproceedings{reimers-2019-sentence-bert,
    title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
    author = "Reimers, Nils and Gurevych, Iryna",
    booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
    month = "11",
    year = "2019",
    publisher = "Association for Computational Linguistics",
    url = "https://arxiv.org/abs/1908.10084",
}

MultipleNegativesRankingLoss

@misc{henderson2017efficient,
    title={Efficient Natural Language Response Suggestion for Smart Reply},
    author={Matthew Henderson and Rami Al-Rfou and Brian Strope and Yun-hsuan Sung and Laszlo Lukacs and Ruiqi Guo and Sanjiv Kumar and Balint Miklos and Ray Kurzweil},
    year={2017},
    eprint={1705.00652},
    archivePrefix={arXiv},
    primaryClass={cs.CL}
}