checkpoint-60/README.md

metadata

tags:
  - sentence-transformers
  - sentence-similarity
  - feature-extraction
  - generated_from_trainer
  - dataset_size:153
  - loss:CoSENTLoss
base_model: sentence-transformers/all-mpnet-base-v2
widget:
  - source_sentence: >-
      1964: Born on July 26, 1964, in Arlington County, Virginia | 1982: attends
      at Washington-Liberty High School | 1982: Graduated from Washington-Lee
      High School | 1987: Moved to Manhattan, New York | 1987: obtains a BFA in
      dramatic art from East Carolina University in Greenville | 1989: Cameos in
      Bionic Showdown: The Six Million Dollar Man and the Bionic Woman | 1993:
      gets the role of Lieutenant Lenina Huxley in Demolition Man | 1994: plays
      a bus driver in Speed | 1994: Lead role in Speed | 1995: Headlined While
      You Were Sleeping | 1995: leads in Love at First Sight | 1995: Won Best
      Actress at the 21st Saturn Awards | 1996: Starred in A Time to Kill |
      1996: is in the cast of The Right to Kill?
    sentences:
      - >-
        1974: Born in Oakland, California | 1996: Attended Saint Mary's College
        of California | 2000: Converted to Islam and changed surname from
        Gilmore to Ali | 2000: Graduated from New York University with an MFA |
        2001: Became an Ahmadiyya Muslim | 2001: Began career on TV series
        Crossing Jordan | 2008: First major film role in The Curious Case of
        Benjamin Button | 2016: Won Academy Award for Best Supporting Actor for
        Moonlight | 2017: First child born (daughter) | 2018: Won Academy Award
        for Best Supporting Actor for Green Book | 2019: Starred in the third
        season of True Detective | 2020: Nominated for Primetime Emmy Awards for
        performances in True Detective and Ramy | 2023: Signed Artists4Ceasefire
        letter for Gaza ceasefire | 2025: Called for release of Palestinian
        student activist Mahmoud Khalil
      - >-
        1966: Born in Paris, France | 1966: Born on November 23 in Paris 15e |
        1995: Debut in La Haine, received two César Award nominations | 1995:
        Debut international with La Haine | 1995: Married to Monica Bellucci,
        two daughters Deva and Léonie | 1995: First César Award nomination for
        best new actor | 1996: Met Monica Bellucci on set of The Apartment |
        1999: Married Monica Bellucci in Monaco | 2004: Daughter Deva born |
        2007: Critical acclaim for performance in Eastern Promises | 2008: Won
        César Award for Best Actor for role as Jacques Mesrine | 2009: Began his
        career with À Deriva | 2009: Won César Award for Best Actor for his role
        as Jacques Mesrine | 2010: Daughter Léonie born
      - >-
        1979: Born in Emmett, Idaho | 1996: Participated in International
        Modeling and Talent Association competition | 1997: Appeared on The
        Price Is Right | 1997: Graduated from Centennial High School in Boise,
        Idaho | 2008: Became known for role as Jesse Pinkman in Breaking Bad |
        2009: Starred in The Last House on the Left | 2010: Nominated for
        Primetime Emmy Award for Outstanding Supporting Actor in a Drama Series
        | 2012: Engaged to Lauren Parsekian in Paris | 2013: Declared October 1
        as Aaron Paul Sturtevant Day | 2013: Got Breaking Bad tattoos with Bryan
        Cranston | 2013: Helped organize a contest to raise $1.6 million for
        anti-bullying organization | 2013: Married Lauren Parsekian in Malibu |
        2018: Daughter Story Annabelle born | 2022: Dropped birth surname to
        'Paul'
  - source_sentence: >-
      1989: Born on 23 July in Queen Charlotte's and Chelsea Hospital, London |
      2000: Auditioned for the role of Harry Potter in the film adaptation of
      Harry Potter and the Philosopher's Stone | 2001: Debut in the film 'The
      Tailor of Panama' | 2001: Received a seven-figure salary for the lead role
      in 'Harry Potter and the Philosopher's Stone' | 2002: Starred in 'Harry
      Potter and the Chamber of Secrets' | 2003: Offers a strand of hair and an
      autograph for the British Charity | 2004: Starred in 'Harry Potter and the
      Prisoner of Azkaban' | 2005: Starred in 'Harry Potter and the Goblet of
      Fire' | 2007: Writes in the magazine Rubbish | 2007: Leaves his hand and
      footprints in front of the Grauman Chinese Theatre | 2007: Signed on for
      the final Harry Potter films | 2007: Stars in the TV film Son Jack | 2008:
      Auctioned jeans for Broadway Cares/Equity Fights AIDS | 2008: Nominated
      for a Drama Desk Award
    sentences:
      - >-
        1962: Born on July 3 in Syracuse, New York | 1980: In the early 1980s,
        Tom Cruise has a romance with Melissa Gilbert, then Heather Locklear. |
        1980: Education: Graduated from Shattuck–St. Mary's School in 1980. |
        1981: Obtains his first leading role in Taps | 1983: Appears in American
        Teenagers, Risky Business, and The Outsiders | 1983: Meets Rebecca De
        Mornay on the set of Risky Business. | 1986: Consolidation with Top Gun
        | 1986: Converted to Scientology by Mimi Rogers | 1986: Becomes famous
        with Top Gun | 1986: First major role in The Color of Money | 1986:
        Inaugurates his star on the Walk of Fame | 1986: Member of the Church of
        Scientology International since 1986 | 1986: Wears Aviator Ray-Ban
        glasses | 1986: First role in The Color of Money
      - >-
        1962: Born on July 3 in Syracuse, New York | 1980: Education: Graduated
        from Shattuck–St. Mary's School in 1980. | 1986: Converted to
        Scientology by Mimi Rogers | 1986: First major role in The Color of
        Money | 1987: Married Mimi Rogers on May 9, 1987 | 1988: Starred in Rain
        Man | 1989: Portrayed Ron Kovic in 'Born on the Fourth of July' | 1990:
        Married Nicole Kidman on December 24, 1990 | 1990: Met Nicole Kidman on
        set of Days of Thunder | 1990: Starred in 'Days of Thunder' | 1992:
        Publicly announced following Scientology | 1994: Starred in 'Interview
        with the Vampire' | 1996: Starred as Ethan Hunt in 'Mission: Impossible'
        | 1999: Marty Rathbun was sent to convince him to return to the Church
      - >-
        2001: Graduated from York University in 2001 with a BFA in theatre. |
        2004: Starred in the comedy film Mean Girls as Regina George. | 2004:
        Starred in The Notebook with Ryan Gosling | 2005: Starred in Red Eye
        with Cillian Murphy | 2005: Starred in the romantic comedy Wedding
        Crashers, the psychological thriller Red Eye, and the comedy-drama The
        Family Sto | 2005: Starred in Wedding Crashers with Owen Wilson, Vince
        Vaughn, and Bradley Cooper | 2005: volunteered in Biloxi, Mississippi,
        and Louisiana for clean-up following Hurricane Katrina | 2005: Won MTV
        Movie Award and four Teen Choice Awards | 2006: One-off stage appearance
        in The Vagina Monologues at St. Lawrence Centre for the Arts, Toronto |
        2006: Received a Rising Star Award nomination from the British Academy
        of Film and Television Arts | 2006: Turned down roles in The Devil Wears
        Prada, Casino Royale, Mission: Impossible III, Iron Man, and Get Smart
        (2006-2008) | 2007: sat on a TreeHugger/Live Earth judging panel | 2008:
        Starred in Married Life with Pierce Brosnan and Chris Cooper | 2009:
        Starred in Sherlock Holmes with Robert Downey, Jr. and Jude Law
  - source_sentence: >-
      1964: Born on July 26, 1964, in Arlington County, Virginia | 1982:
      Graduated from Washington-Lee High School | 1987: Moved to Manhattan, New
      York | 1989: Cameos in Bionic Showdown: The Six Million Dollar Man and the
      Bionic Woman | 1994: Lead role in Speed | 1995: Headlined While You Were
      Sleeping | 1995: Won Best Actress at the 21st Saturn Awards | 1996:
      Starred in A Time to Kill | 1998: Starred in Hope Floats | 1998: Voiced
      Miriam in The Prince of Egypt | 1999: Starred in Forces of Nature | 2000:
      Private jet crash on a runway | 2000: Starred in 28 Days | 2000: Starred
      in Miss Congeniality
    sentences:
      - >-
        1983: Born on 11 August in Melbourne, Victoria | 1983: Born on August 11
        in Melbourne (Victoria) | 2003: Begins in Summer Bay | 2004: Became a
        part of Home and Away | 2007: Left Home and Away | 2009: Played James T.
        Kirk's father in Star Trek | 2010: Portrays Sam Phelan in Ca$h | 2010:
        Married Elsa Pataky | 2011: Portrays Thor | 2011: Starred as Thor in the
        Marvel Cinematic Universe | 2012: Daughter India Rose born | 2012:
        Portrays the male lead in the fantasy blockbuster Snow White and the
        Huntsman | 2012: Portrays Curt Vaughan in The Cabin in the Woods | 2012:
        Reteams with Joss Whedon for Avengers
      - >-
        1970: Born on 7 March in Westminster, London | 1988: Graduated from
        Trinity Hall, Cambridge with upper second-class honours | 1994: Stage
        debut in 'Design for Living' | 1999: International breakthrough in 'The
        Mummy' | 2001: Began dating Darren Aronofsky | 2004: Appeared in the
        comedy Envy | 2005: Starred in The Constant Gardener | 2005: Won
        Laurence Olivier Award for Best Actress for 'A Streetcar Named Desire' |
        2006: Received BAFTA Britannia Award for British Artist of the Year |
        2006: Starred in The Fountain | 2009: Played Hypatia of Alexandria in
        Agora | 2010: Starred in The Whistleblower | 2011: Became a naturalised
        American citizen | 2011: Played Blanche DuBois in A Streetcar Named
        Desire
      - >-
        1970: Born on May 25 in Montgomery, Alabama | 1988: died her mother |
        1988: Graduated from Jefferson Davis High School | 1996: Made film debut
        as a nurse in A Time to Kill | 1996: meets Tate Taylor | 1996: moves to
        Los Angeles | 1996: her first role in The Right Stuff | 1998: lands a
        role in The Help | 2007: participates in the series Ugly Betty | 2011:
        lands the role of Aibileen in The Help | 2011: wins the Oscar for Best
        Supporting Actress | 2012: Won Academy Award for Best Supporting Actress
        for The Help | 2012: Won BAFTA for Best Supporting Actress for The Help
        | 2013: Won National Board of Review Award for Best Supporting Actress
        for Fruitvale Station
  - source_sentence: >-
      1988: born on November 6 in Scottsdale (Arizona) | 2004: Debut on
      television in In Search of the New Partridge Family | 2004: made her
      acting debut in 2004 | 2007: Feature film debut in Superbad | 2009: Moved
      from Los Angeles to Greenwich Village, New York | 2009: got her first
      major role in the comedy Easy Girl | 2009: Starred in Paper Man | 2009:
      Starred in Zombieland | 2009: lived in Los Angeles for 5 years | 2010:
      dated Kieran Culkin (1 year) | 2010: Hosted Saturday Night Live | 2010:
      Voice role in Marmaduke | 2011: joined the cast of La La Land | 2011:
      Starred in Crazy, Stupid, Love and The Help
    sentences:
      - >-
        1970: Born on May 25 in Montgomery, Alabama | 1988: died her mother |
        1988: Graduated from Jefferson Davis High School | 1996: Made film debut
        as a nurse in A Time to Kill | 1996: meets Tate Taylor | 1996: moves to
        Los Angeles | 1996: her first role in The Right Stuff | 1998: lands a
        role in The Help | 2007: participates in the series Ugly Betty | 2011:
        lands the role of Aibileen in The Help | 2011: wins the Oscar for Best
        Supporting Actress | 2012: Won Academy Award for Best Supporting Actress
        for The Help | 2012: Won BAFTA for Best Supporting Actress for The Help
        | 2013: Won National Board of Review Award for Best Supporting Actress
        for Fruitvale Station
      - >-
        2004: Debut on television in In Search of the New Partridge Family |
        2007: Feature film debut in Superbad | 2009: Moved from Los Angeles to
        Greenwich Village, New York | 2009: Starred in Paper Man | 2009: Starred
        in Zombieland | 2010: Hosted Saturday Night Live | 2010: Voice role in
        Marmaduke | 2011: Starred in Crazy, Stupid, Love and The Help | 2011:
        Starred in Friends with Benefits | 2013: Amazing Actress | 2014:
        Breakout Movie Star | 2014: Reprised role of Gwen Stacy in The Amazing
        Spider-Man 2 | 2014: Starred in Birdman | 2014: Starred in Woody Allen's
        romantic comedy Magic in the Moonlight
      - >-
        1979: Born in Guatemala City | 2005: Graduated from Juilliard | 2005:
        Played Proteus in Two Gentlemen of Verona | 2006: First major role in
        The Nativity Story | 2007: Engaged to Maria Miranda in 2007 | 2007:
        Played Romeo in Romeo and Juliet | 2009: Won AACTA Award for Best Actor
        in a Supporting Role | 2009: Won AACTA Award for Best Actor in a
        Supporting Role for Balibo | 2010: Played King John in Robin Hood |
        2011: Played a security guard in Madonna's W.E. | 2012: Nominated for
        ALMA Award for Favorite Movie Actor – Supporting Role for For Greater
        Glory | 2013: Nominated for Golden Globe Award for Best Actor – Motion
        Picture Musical or Comedy for Inside Llewyn Davis | 2014: Starred in A
        Most Violent Year | 2014: Starred in The Two Faces of January
  - source_sentence: >-
      1983: Born on 11 August in Melbourne, Victoria | 2004: Became a part of
      Home and Away | 2007: Left Home and Away | 2009: Played James T. Kirk's
      father in Star Trek | 2010: Married Elsa Pataky | 2011: Starred as Thor in
      the Marvel Cinematic Universe | 2012: Daughter India Rose born | 2012:
      Played the character Kale Garrity in A Perfect Getaway | 2012: Reprised
      the role of Thor in The Avengers | 2013: Reprised Thor in Thor: The Dark
      World (2013) | 2013: Starred in Rush | 2014: Donated fees to the
      Australian Children Foundation | 2014: Twin sons born | 2015: Co-starred
      in Vacation
    sentences:
      - >-
        2004: Debut on television in In Search of the New Partridge Family |
        2007: Feature film debut in Superbad | 2009: Moved from Los Angeles to
        Greenwich Village, New York | 2009: Starred in Paper Man | 2009: Starred
        in Zombieland | 2010: Hosted Saturday Night Live | 2010: Voice role in
        Marmaduke | 2011: Starred in Crazy, Stupid, Love and The Help | 2011:
        Starred in Friends with Benefits | 2013: Amazing Actress | 2014:
        Breakout Movie Star | 2014: Reprised role of Gwen Stacy in The Amazing
        Spider-Man 2 | 2014: Starred in Birdman | 2014: Starred in Woody Allen's
        romantic comedy Magic in the Moonlight
      - >-
        1970: Born on 7 March in Westminster, London | 1983: education at North
        London Collegiate School | 1986: education at Benenden School | 1988:
        Graduated from Trinity Hall, Cambridge with upper second-class honours |
        1990: studies at Trinity Hall, University of Cambridge | 1994: Stage
        debut in 'Design for Living' | 1999: lands the role of Evelyn Carnahan
        in The Mummy | 1999: International breakthrough in 'The Mummy' | 1999:
        lives with Sam Mendes | 2001: Began dating Darren Aronofsky | 2004:
        Appeared in the comedy Envy | 2005: Starred in The Constant Gardener |
        2005: Won Laurence Olivier Award for Best Actress for 'A Streetcar Named
        Desire' | 2006: Received BAFTA Britannia Award for British Artist of the
        Year
      - >-
        1988: Born in Paris | 2016: First breakthrough in Heaven Will Wait |
        2019: Starred in Portrait of a Lady on Fire | 2019: Won Lumière Award
        for Best Actress | 2021: Made feature directorial debut with Mi Iubita
        Mon Amour | 2022: Portrayed Francesca Lentini in Tár | 2023: Announced
        to play Emmanuelle in reboot | 2023: Invited to become a member of the
        Academy of Motion Picture Arts and Sciences | NA: Parents are real
        estate agents | NA: Raised in Rezé near Nantes
pipeline_tag: sentence-similarity
library_name: sentence-transformers
metrics:
  - pearson_cosine
  - spearman_cosine
model-index:
  - name: SentenceTransformer based on sentence-transformers/all-mpnet-base-v2
    results:
      - task:
          type: semantic-similarity
          name: Semantic Similarity
        dataset:
          name: validation
          type: validation
        metrics:
          - type: pearson_cosine
            value: 0.9414523862441698
            name: Pearson Cosine
          - type: spearman_cosine
            value: 0.6996947820398319
            name: Spearman Cosine

SentenceTransformer based on sentence-transformers/all-mpnet-base-v2

This is a sentence-transformers model finetuned from sentence-transformers/all-mpnet-base-v2. 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: sentence-transformers/all-mpnet-base-v2
• Maximum Sequence Length: 384 tokens
• Output Dimensionality: 768 dimensions
• Similarity Function: Cosine Similarity

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': 384, 'do_lower_case': False}) with Transformer model: MPNetModel 
  (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("sentence_transformers_model_id")
# Run inference
sentences = [
    "1983: Born on 11 August in Melbourne, Victoria | 2004: Became a part of Home and Away | 2007: Left Home and Away | 2009: Played James T. Kirk's father in Star Trek | 2010: Married Elsa Pataky | 2011: Starred as Thor in the Marvel Cinematic Universe | 2012: Daughter India Rose born | 2012: Played the character Kale Garrity in A Perfect Getaway | 2012: Reprised the role of Thor in The Avengers | 2013: Reprised Thor in Thor: The Dark World (2013) | 2013: Starred in Rush | 2014: Donated fees to the Australian Children Foundation | 2014: Twin sons born | 2015: Co-starred in Vacation",
    "1970: Born on 7 March in Westminster, London | 1983: education at North London Collegiate School | 1986: education at Benenden School | 1988: Graduated from Trinity Hall, Cambridge with upper second-class honours | 1990: studies at Trinity Hall, University of Cambridge | 1994: Stage debut in 'Design for Living' | 1999: lands the role of Evelyn Carnahan in The Mummy | 1999: International breakthrough in 'The Mummy' | 1999: lives with Sam Mendes | 2001: Began dating Darren Aronofsky | 2004: Appeared in the comedy Envy | 2005: Starred in The Constant Gardener | 2005: Won Laurence Olivier Award for Best Actress for 'A Streetcar Named Desire' | 2006: Received BAFTA Britannia Award for British Artist of the Year",
    '1988: Born in Paris | 2016: First breakthrough in Heaven Will Wait | 2019: Starred in Portrait of a Lady on Fire | 2019: Won Lumière Award for Best Actress | 2021: Made feature directorial debut with Mi Iubita Mon Amour | 2022: Portrayed Francesca Lentini in Tár | 2023: Announced to play Emmanuelle in reboot | 2023: Invited to become a member of the Academy of Motion Picture Arts and Sciences | NA: Parents are real estate agents | NA: Raised in Rezé near Nantes',
]
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]

Evaluation

Metrics

Semantic Similarity

• Dataset: validation
• Evaluated with EmbeddingSimilarityEvaluator

Metric	Value
pearson_cosine	0.9415
spearman_cosine	0.6997

Training Details

Training Dataset

Unnamed Dataset

• Size: 153 training samples
• Columns: sentence1, sentence2, and score
• Approximate statistics based on the first 153 samples:

  	sentence1	sentence2	score
  type	string	string	int
  details	min: 109 tokens mean: 159.55 tokens max: 251 tokens	min: 109 tokens mean: 162.8 tokens max: 251 tokens	0: ~73.86% 1: ~26.14%

• Samples:

  sentence1	sentence2	score
  1979: Born in Virginia, Minnesota on June 21	1997: Graduated from Lake Stevens High School	2000: Cursed Part 3 (short horror film)
  1967: Born in Shirebrook, Derbyshire, England on 26 July	1978: Attended local grammar school	1978: Begins playing football in high school
  1989: Born on 23 July in Queen Charlotte's and Chelsea Hospital, London	2000: Auditioned for the role of Harry Potter in the film adaptation of Harry Potter and the Philosopher's Stone	2001: Debut in the film 'The Tailor of Panama'

• Loss: CoSENTLoss with these parameters:

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

Evaluation Dataset

Unnamed Dataset

• Size: 39 evaluation samples
• Columns: sentence1, sentence2, and score
• Approximate statistics based on the first 39 samples:

  	sentence1	sentence2	score
  type	string	string	int
  details	min: 109 tokens mean: 156.23 tokens max: 208 tokens	min: 109 tokens mean: 158.18 tokens max: 251 tokens	0: ~79.49% 1: ~20.51%

• Samples:

  sentence1	sentence2	score
  1970: Born on May 25 in Montgomery, Alabama	1988: Graduated from Jefferson Davis High School	1996: Made film debut as a nurse in A Time to Kill
  1964: Born on July 26, 1964, in Arlington County, Virginia	1982: Graduated from Washington-Lee High School	1987: Moved to Manhattan, New York
  1980: Born in London, Ontario, Canada	1993: Joined The All New Mickey Mouse Club	1995: Returned to Canada

• Loss: CoSENTLoss with these parameters:

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

Training Hyperparameters

Non-Default Hyperparameters

• eval_strategy: epoch
• per_device_eval_batch_size: 16
• learning_rate: 3e-05
• warmup_ratio: 0.1
• fp16: True
• load_best_model_at_end: True
• ddp_find_unused_parameters: False

All Hyperparameters

Click to expand

• overwrite_output_dir: False
• do_predict: False
• eval_strategy: epoch
• prediction_loss_only: True
• per_device_train_batch_size: 8
• per_device_eval_batch_size: 16
• 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: 3e-05
• weight_decay: 0.0
• adam_beta1: 0.9
• adam_beta2: 0.999
• adam_epsilon: 1e-08
• max_grad_norm: 1.0
• num_train_epochs: 3
• max_steps: -1
• lr_scheduler_type: linear
• lr_scheduler_kwargs: {}
• warmup_ratio: 0.1
• 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: True
• 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: True
• 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}
• 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: False
• 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
• dispatch_batches: None
• split_batches: 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	validation_spearman_cosine
0.15	3	0.2656	-	-
0.3	6	0.0915	-	-
0.45	9	0.0079	-	-
0.6	12	0.0004	-	-
0.75	15	0.0006	-	-
0.9	18	0.0082	-	-
1.0	20	-	0.0018	0.6997
1.05	21	0.0002	-	-
1.2	24	0.0006	-	-
1.35	27	0.0003	-	-
1.5	30	0.0022	-	-
1.65	33	0.0007	-	-
1.8	36	0.0002	-	-
1.95	39	0.0	-	-
2.0	40	-	0.0005	0.6997
2.1	42	0.0	-	-
2.25	45	0.0	-	-
2.4	48	0.0	-	-
2.55	51	0.0002	-	-
2.7	54	0.0	-	-
2.85	57	0.0	-	-
3.0	60	0.0001	0.0003	0.6997

Framework Versions

• Python: 3.10.16
• Sentence Transformers: 3.3.1
• Transformers: 4.48.0
• PyTorch: 2.4.0
• Accelerate: 1.2.1
• Datasets: 3.2.0
• Tokenizers: 0.21.0

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",
}

CoSENTLoss

@online{kexuefm-8847,
    title={CoSENT: A more efficient sentence vector scheme than Sentence-BERT},
    author={Su Jianlin},
    year={2022},
    month={Jan},
    url={https://kexue.fm/archives/8847},
}