Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks
Paper • 1908.10084 • Published • 15
How to use ahmedHamdi/IR-pt-en-gemma with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("ahmedHamdi/IR-pt-en-gemma")
sentences = [
"In the Victorian era, the eminent Doctor Sir Colenso Ridgeon announces that he has created an effective serum against the deadly tuberculosis, and his doctor friends go to his house to honor him. After the celebration, Colenso's maid announces that a woman, Jennifer Dubedat, is seeking him out because her husband, the young artist Louis, suffers from the disease. The doctor initially refuses to treat her, but eventually succumbs to Jennifer's charm and agrees to include Louis in his small group of patients. But when they meet Louis, the doctors discover that he is a swindler and bigamist, deceiving Jennifer. Then the humble doctor, Dr. Blenkinsop, who has no notoriety and successfully treats his poor patients with natural remedies, tells Colenso that he too is tubercular. The doctor then faces a dilemma: since he can no longer accept patients, he is unsure whether it is Louis, whom he detests, or his old friend Dr. Blenkinsop.",
"In London in 1906, some colleagues rush round to the Harley Street house of the unmarried Dr. Ridgeon to congratulate him on being awarded a knighthood for his claim that he can cure tuberculosis (though in reality at that time there was no certain cure). Among them is Dr. Blenkinsop, who was at medical school with him, but now looks after poor patients who cannot afford to pay much, and is himself infected with tuberculosis. Waiting in his hall is the young and lovely Jennifer Dubedat who, having heard of his alleged cure, wants him to treat her husband Louis, a penniless artist. Struck by the beauty and charm of Jennifer, and by the quality of her husband's drawings which she shows him, he lets her believe he will try to save Louis, but stresses that his treatment is long and expensive, and that he can only handle ten patients at a time. So that he can meet Louis, and see more of Jennifer, he invites the two to a celebration dinner he is hosting that night. While Louis is like his wife good-looking and charming, it emerges that he scorns all trappings of conventional morality, being a feckless liar, thief and seducer (a waitress recognises him as her vanished husband). Now highly doubtful about Louis, Ridgeon arranges with colleagues to visit the studio where he lives and assess his case. Recognising that his TB is far advanced, Ridgeon passes him on to a colleague and decides to take Blenkinsop instead. To Jennifer this is treachery and she begins to hate Ridgeon, while he hopes that he might marry her as soon as she is a widow. Beyond cure, Louis soon dies and in a moving last speech states his belief that love and beauty surpass all of conventional morality. He also begs Jennifer not to mourn, but to marry again and be happy. In a postscript, Ridgeon attends the one-man exhibition of Louis' works, which are selling well. Jennifer refuses to let him buy a painting and leaves with her new husband.",
"In 1914, young Sara Crewe lives in India with her widower father, Captain Richard Crewe, a wealthy British Army Officer, sharing his love for stories of myths and magic. Called to serve in the Great War, Richard enrolls her at an all-girls' boarding school in New York City, which her late mother had attended. Run by the haughty headmistress, Miss Maria Minchin, and her kindly sister, Amelia. Instructing Miss Minchin to spare no expense for his daughter's comfort, Richard furnishes the school's largest suite and leaves Sara with her mother's locket and a doll he names Emily, which he tells her will keep them connected through magic. Although stifled by Miss Minchin's strictness, Sara becomes popular among the girls, including the scullery maid, Becky, for her kindness and powerful imagination. Sara writes to her father, who is caught in a gas attack while trying to save a fellow soldier in the trenches. Miss Minchin throws Sara a lavish birthday party, hoping to extort more money from her father, but his solicitor arrives with news that Richard has been killed in battle; the British government has seized his assets, leaving Sara penniless. Miss Minchin moves Sara to the attic with Becky to work as a servant and confiscates her belongings, including the locket, allowing her to keep only Emily and a book. Though her life is bleak, Sara remains kind to others, but gets her revenge on the school bully, Lavinia. Charles Randolph, the school's elderly neighbor, receives word that his son John has been declared missing in action while fighting in Europe. Ram Dass, Mr. Randolph's Indian associate, comes to notice Sara from the neighboring attic, overhearing her imaginative stories. When a wounded soldier suffering from amnesia is misidentified as John, Ram Dass encourages Mr. Randolph to take the man in. Sara's friends sneak into Miss Minchin's office and recover her locket, visiting her that night to hear her tales of Prince Rama. They are caught by Miss Minchin, who punishes Sara and Becky with a day without meals and keeping Becky locked in her room for the entire day, but Sara stands up to her cruelty with her father's belief that all girls are princesses no matter their lot in life. Sara comforts Becky by imagining a feast and fine clothes for them, and they awaken to find the dream has come true, with their attic secretly transformed by Ram Dass. Inspired by Sara, Amelia runs away with the milkman, and Miss Minchin discovers the locket is missing. Confronting Sara in the attic, Miss Minchin accuses her of stealing the finery left by Ram Dass and she locks Sara in her room while she summons the police. With Becky's help, Sara narrowly escapes by making the perilous climb over to the Randolph house. As Miss Minchin and the police search the house, Sara discovers the recovering soldier is none other than her father, but Richard does not recognize her. Though Miss Minchin does clearly recognize Captain Crewe, she deliberately chooses revenge over the truth by spitefully saying that Sara has no father. As Sara is dragged away by the police screaming for her father, Ram Dass helps Richard regain his memory. Outside Richard saves Sara and the two are happily reunited while Miss Minchin angrily walks away, finally accepting defeat. Sometime later, Mr. Randolph has taken over the school, now a much happier place for the girls, and has found peace in knowing that Richard tried to save his son. Richard's fortune has been restored and he has adopted Becky. As punishment for her vile treatment to the girls and other crimes, Miss Minchin is reduced to working for a young chimney sweep she mistreated earlier. Sara gives Emily to the girls and shares an unexpected hug with Lavinia, before she and Becky depart for home.",
"In the far future, three humans—Jirou, Hal, and Maria—discover that an alien race called the Pasateli intends to conquer humankind with the mysterious Cleopatra Plan. Using a time machine, the three transport their minds into the bodies of members of the historical Cleopatra's court to discover and stop the plan. Hal, however, vows to use the opportunity to secure the title of the greatest lover who ever lived by having sex with Cleopatra. In the middle of the Roman conquest of Egypt, a group of Egyptians secretly plotted a rebellion to overthrow Julius Caesar. The group plans to send Cleopatra to seduce and murder Caesar. The Romans discovered the group and attacked them. Cleopatra escapes, along with her handmaidens Libya and Apollodoria. Cleopatra goes to an ancient priestess, who magically grants her an irresistibly seductive body for her mission. It is at this point when Jirou, Hal, and Maria arrive: Maria is now Libya; Hal finds himself in the body of the priestess' pet leopard Rupa, thwarting his plans to seduce Cleopatra; and Jirou finds himself in the body of Ionius, a Greek man captured and enslaved by the Romans. Ionius frees himself and the other slaves by using his knowledge of future technology to make modern hand grenades. They accompany Cleopatra to meet Caesar, who is so overcome by her beauty that he makes her queen of Egypt. Caesar recaptures Ionius and, amused by his fighting skills, orders him to fight in the gladiatorial arena. He gives Ionius a modern-day handgun to ensure his victory. Ionius proves so popular with the Roman public that Caesar's own popularity soars, leading his senators to conspire to murder both him and Ionius to end their influence. Libya and Apollodoria insist Cleopatra must murder Caesar; Cleopatra, however, has had a change of heart and keeps putting off the assassination in favor of sex. They accompany Caesar back to Rome, just in time for him to be assassinated by his own senators. Caesar's adopted son, Octavian—soon to be called Augustus—takes command. Cleopatra tries to continue the plan by seducing Octavian, only to learn that he is homosexual and impervious to her charms. Meanwhile, Caesar's right-hand man, Marcus Antonius or Anthony, falls in love and has sex with Cleopatra. Octavian, on the other hand, is attracted to Ionius and spares his life. Finally, during the Battle of Actium where Octavian's fleet defeats Anthony's Egyptian fleet, Anthony kills himself. Octavian goes to Cleopatra to persuade her to surrender and is taken into custody by the Romans. Distraught and disappointed by the rejection after Anthony's death, Cleopatra commits suicide using an asp's venomous bite. The time travelers return to the future and report that the Cleopatra Plan is a scheme by the Pasateli to assume the form of beautiful human women to seduce and destroy Earth's most powerful male leaders. The Pasateli have already taken their human forms and are poised to strike when this information arrives, but Earth is able to root them out and save the world in time."
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]This is a sentence-transformers model finetuned from google/embeddinggemma-300m. 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.
SentenceTransformer(
(0): Transformer({'max_seq_length': 256, 'do_lower_case': False, 'architecture': 'Gemma3TextModel'})
(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})
)
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("ahmedHamdi/IR-pt-en-gemma")
# Run inference
sentences = [
"Tom Brand (Kevin Spacey) is a business tycoon who has separated his daughter Rebecca (Malina Weissman) and his wife Lara (Jennifer Garner). In a rush to get a last-minute gift for his daughter's birthday, he goes to a mysterious pet shop run by Felix Grant (Christopher Walken) and buys a cat. On the way home, Tom is involved in an accident and becomes trapped inside the cat's body. Felix tells him that he must reconcile with his family within a week or he will remain a cat forever.",
"Tom Brand is a major business tycoon in New York City whose workaholic attitude ruined his first marriage with Madison, and his adult son David now works for him and strives for his approval. Tom and David now live with their second wife and stepmother Lara, who is a little more tolerant of the fact that Tom is never at home, and their daughter and half-sister Rebecca. Tom’s eponymous company FireBrand is nearing completion on its greatest achievement to date: the tallest skyscraper in the northern hemisphere that will be the new headquarters. In the opening scene, he skydives out of a plane and lands on the top of the new skyscraper, while David remains in the plane, refusing to jump. Rebecca's 11th birthday comes up and she has always wanted a cat, but Tom has always refused as he hates cats. Not wanting to disappoint her, he rushes to get a last-minute present for his daughter's birthday, but as he is distracted, the GPS redirects him to a mysterious pet store called Purrkins Pet Shop, brimming with odd and exotic cats. The store's eccentric owner Felix Perkins tells him he does not pick the cat, the cat picks him. The tomcat that picks him is called Mr. Fuzzypants, who, Perkins says, has used up seven of his nine lives. Tom buys the cat. On his way home for the party, Tom meets with Ian Cox, one of the top managers of the company in charge of the new building, and learns that another building in Chicago will be taller. They have an argument while on the roof that ends with Tom firing Ian, but lightning strikes the antenna. Tom is blown off the building along with the cat, and without any effort from Ian to save him, he plummets off the side of the skyscraper, but his leg gets snagged on some stray equipment and gets flung back inside through a window as he passes out. When he wakes up, he realizes that his human body is in the hospital in a state of coma, and his consciousness is trapped inside the cat's body. Tom is visited by Felix, who is able to talk to him and knows what happened. Felix warns Tom to reevaluate his priorities, connect with his family, and avoid past mistakes within one week or else remain as a cat forever. When Lara and Rebecca take Mr. Fuzzypants home, he acts in an odd and stubborn way to try to convince his wife and daughter that he is actually Tom. This only drives them mad and he slowly comes to see how much he has ignored his family. He suspects that Lara is cheating on him with model Josh Myers and they had been looking at houses together as a prelude to divorcing him. This motivates him to try to make her happy. He also learns that Ian is trying to make the company public with the help of the board of directors to take power from Tom, even though David is trying to stop him. Ian fires David from the company and plans to announce that the company will be sold at the party for the new tower opening. Eventually, Rebecca realizes that Mr. Fuzzypants is really her father. Meanwhile, Tom's body is in crisis at the hospital and Lara, David, and Rebecca all go there. Rebecca has the cat hidden in her backpack. In a moment alone, David apologizes to Tom for failing to save the company and takes Tom's ID badge, implying he will commit suicide at the tower. At the hospital, Lara and Dr. Cole plan for Lara to sign a DNR, which is a legal form, indicating that if Tom's heart stops beating, they won't save him. Rebecca calls to the cat to come prove he is really Tom, but Tom remembers what Felix said about love being sacrifice and decides to save David instead, sacrificing his only chance at regaining his humanity. At the tower, David jumps off the building and the cat jumps after him pulling a cable. It is then revealed that David is wearing a BASE jumping parachute. He lands in the middle of the party, presents the articles of incorporation of the company, and announces that he now controls his father's 51% of the stock. He says the company will remain a family company and fires Ian. Although the cat is not seen landing, Tom wakes up from his coma just in time to see David's announcement on the television. Meanwhile, Ian passes Felix who tells him to hang up his cell phone. Ian ignores him, but is then hit by a car, and his consciousness transfers into a cat that Felix takes back to his shop. Tom and Rebecca return to Felix where Tom asks if he has any dogs. Felix says he does not, but presents Mr. Fuzzypants, who has one life left.",
"Aging Greek Theo Tomasis rose from his peasant roots to become a mogul who owns oil tankers, airlines, and Mediterranean islands and longs to be elected President of Greece. He eventually falls in love with the considerably younger Liz Cassidy. The two first meet when she is visiting his island estate with her husband James, the Senator from the state of Massachusetts. Theo immediately is attracted to her and, despite the fact she is happily married, begins to woo her aboard his yacht while James converses with the former Prime Minister of the United Kingdom. Meanwhile, Theo's relationship with his son, Nico, is tenuous at best. He encourages Nico to join the business and make money in the same manner he did. Nico wants to manage his own affairs and convinces his father to jumpstart his ventures. The plan is to buy older freighters and convert them to oil tankers and sell them to the Saudi family. This would allow Saudi oil magnates to control the delivery of oil outside of American Corporate controls. Theo is impressed and helps bankroll the purchase of a fleet of boats. Upon their return to the States, the Cassidys begin their run for President of the United States, eventually winning the primary and the general election. The First Lady later suffers a miscarriage and becomes depressed and tired of the public life. Tomasis, who has a good relationship with the First Family, offers a vacation on his yacht as a change of pace. The President does not care for Tomasis due to his shady dealings in the shipping business and Arab oil states, so Lizzie goes by herself, much to his anger. In Greece, the yacht is full of guests and celebrities. Tomasis seems to spend a lot of time with Lizzie, who is a bit restrained. During a dinner however, Lizzie joins in a traditional sensual dance with plates being thrown and the other guests clapping. Lizzie is attracted to Tomasis' free spirit and joie de vivre and leaves the vacation early to go back to the states-much to his protests. Back in the US, Lizzie and James walk on the family beach-front property, where the President is assassinated by a sharpshooter. After the funeral, Lizzie becomes depressed and stays outside of public eyes even though the former President's brother John (the attorney general) requests she help promote his campaign. She refuses, and Tomasis, re-invites her to Greece for a longer stay. In Greece, the relationship blooms and Lizzie cheers up. Tomasis' business dealings with Nico are becoming scrutinized by American politicians due to the OPEC embargoes and anger over oil prices. Freighters are getting confiscated and the investment of hundreds of millions at risk. Upon his return from the US, Theo asks Lizzie to marry him. Lizzie wants to but is worried about appearances in her country but he outlines the arrangement that is fair and financially attractive enough to abandon the Cassidy family fortune and control. After the wedding, there are growing pains for the relationship, as Lizzie is accustomed to being a voice in her husband's affairs in the past. After Tomasis berates her for interfering in his business, he apologizes for his Greek chauvinism, and they learn to live with each other's cultural differences. The marriage is loving until Niko dies in an accident, and Tomasis gets depressed. Lizzie manages to coax him out of his moods, and things appear to look up as his lawyer and representative cops him a cheap plea deal on the oil manipulation charges. However, his health declines and a doctor gives him a bad prognosis. He holds his medical information secret as his energy and health declines. When Liz wants to shop in Paris, he encourages her to go alone and enjoy herself. She leaves the boat on her vacation, waving, and Tomasis visits his old peasant home on the coast for the last time visiting his old haunts and ending with traditional Greek dance.",
]
embeddings = model.encode(sentences)
print(embeddings.shape)
# [3, 768]
# Get the similarity scores for the embeddings
similarities = model.similarity(embeddings, embeddings)
print(similarities)
# tensor([[1.0000, 0.8029, 0.0491],
# [0.8029, 1.0000, 0.0805],
# [0.0491, 0.0805, 1.0000]])
sentence_0 and sentence_1| sentence_0 | sentence_1 | |
|---|---|---|
| type | string | string |
| details |
|
|
| sentence_0 | sentence_1 |
|---|---|
Steve Grayson (Elvis Presley) is a champion race car driver with a generous soul, sharing all his winnings with people in need. Because of his generosity and his boss's (Bill Bixby) passion for gambling and women, Steve finds himself drowning in debt. |
Steve Grayson (Presley) is a generous NASCAR race car driver with a heart of gold who feels compelled to bail friends and acquaintances out of financial hardship. However, Steve's manager and best friend Kenny Donford, a compulsive gambler, had been mismanaging Steve's winnings to support his gambling habits, landing Steve in deep trouble with the IRS for nonpayment of back taxes and causing many of Steve's valuable possessions to be repossessed. This proves to be a problem for Steve in his efforts to continue racing competitively and support those who depend on his intense generosity. Enter Susan Jacks (Sinatra), an IRS agent assigned to keep tabs on Steve and apply his future prize money toward his $150,000 debt, but she ends up taking a romantic interest in him as well. |
The lives of three New York couples take incredible turns after two of them, Rebecca (Marley Shelton) and Dag (Ron Eldard), share a forbidden night of love. More betrayal, passion, revenge, and death follow the couple's unmasking, but what if nothing more than a kiss had happened that night? |
Dag is a successful director of television commercials who lives with his girlfriend, Halley. Dag, however, has a serious case of wandering eye and is given to frequent flings with other women. Halley tries to turn a blind eye to Dag's infidelity, but when she discovers Dag had a one-night stand with Rebecca, a beautiful but troubled ballet dancer who is dating Dag's close friend Peter, she decides things have gone too far. To add to her feeling of betrayal, she hears a message on the answering machine for Dag from another woman looking to meet up. Halley breaks up with Dag; soon after, she makes the acquaintance of Andre, a handsome and well-mannered classical musician. Andre, however, is married to Colleen, a woman with exotic sexual tastes who meets up with Peter, now suddenly without a girlfriend, on an airline flight. Peter and Colleen have sex in the plane’s bathroom. Just before the plane lands, Peter makes a cell phone call, which causes interference with the signals from the p... |
A horror and suspense thriller following the formula of Steven Spielberg's 1971 film Duel. The peaceful village of Santa Ynez, in the Southwestern United States, is shaken when a mysterious, all-black, extremely fast car runs over and kills people without any apparent reason. The episode unsettles Sheriff Wade Parent (Brolin), Sheriff Everett Peck (Marley), and Deputy Luke Johnson (Cox). The car reappears and kills Everett. An eyewitness, an old Native American woman, claims that the vehicle has no driver, and the assumption that it is a ghost car seems to be confirmed when the car threatens a school parade and Wade's girlfriend, Lauren (Lloyd), manages to lead students and teachers to a cemetery: in confrontation with Christian symbols, the car retreats and flees. The road suspense continues with a thrilling chase, culminating in a direct confrontation between Wade and the Devil's machine. And the plot unfolds with other climaxes until an epilogue." Apocalyptic, with special effects b... |
Two cyclists riding through a canyon are followed down the road by a mysterious matte black car. At a bridge, the car proceeds to crush one cyclist against the wall and ram the other from behind, catapulting him off the bridge. A hitchhiker, hoping to get a ride, encounters the car and insults it after it purposely tries to run him down. It halts to a stop, reverses backwards, and runs him over several times before driving off. The local sheriff's office, called to a series of hit and runs, gets a lead on the car. It appears to be heavily customized and has no license plates, as pointed out by Amos Clemens after he saw it run over the hitchhiker. That night, after intentionally swerving around Amos, the car instead runs over Sheriff Everett Peck, leaving Chief Deputy Wade Parent in charge. During the resulting investigation, an eyewitness to the accident states there was no driver, furthering Wade's confusion. Wade asks his girlfriend, Lauren, a teacher at the local school, to cancel t... |
MultipleNegativesRankingLoss with these parameters:{
"scale": 20.0,
"similarity_fct": "cos_sim",
"gather_across_devices": false
}
per_device_train_batch_size: 4per_device_eval_batch_size: 4multi_dataset_batch_sampler: round_robinoverwrite_output_dir: Falsedo_predict: Falseeval_strategy: noprediction_loss_only: Trueper_device_train_batch_size: 4per_device_eval_batch_size: 4per_gpu_train_batch_size: Noneper_gpu_eval_batch_size: Nonegradient_accumulation_steps: 1eval_accumulation_steps: Nonetorch_empty_cache_steps: Nonelearning_rate: 5e-05weight_decay: 0.0adam_beta1: 0.9adam_beta2: 0.999adam_epsilon: 1e-08max_grad_norm: 1num_train_epochs: 3max_steps: -1lr_scheduler_type: linearlr_scheduler_kwargs: Nonewarmup_ratio: 0.0warmup_steps: 0log_level: passivelog_level_replica: warninglog_on_each_node: Truelogging_nan_inf_filter: Truesave_safetensors: Truesave_on_each_node: Falsesave_only_model: Falserestore_callback_states_from_checkpoint: Falseno_cuda: Falseuse_cpu: Falseuse_mps_device: Falseseed: 42data_seed: Nonejit_mode_eval: Falsebf16: Falsefp16: Falsefp16_opt_level: O1half_precision_backend: autobf16_full_eval: Falsefp16_full_eval: Falsetf32: Nonelocal_rank: 0ddp_backend: Nonetpu_num_cores: Nonetpu_metrics_debug: Falsedebug: []dataloader_drop_last: Falsedataloader_num_workers: 0dataloader_prefetch_factor: Nonepast_index: -1disable_tqdm: Falseremove_unused_columns: Truelabel_names: Noneload_best_model_at_end: Falseignore_data_skip: Falsefsdp: []fsdp_min_num_params: 0fsdp_config: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}fsdp_transformer_layer_cls_to_wrap: Noneaccelerator_config: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}parallelism_config: Nonedeepspeed: Nonelabel_smoothing_factor: 0.0optim: adamw_torch_fusedoptim_args: Noneadafactor: Falsegroup_by_length: Falselength_column_name: lengthproject: huggingfacetrackio_space_id: trackioddp_find_unused_parameters: Noneddp_bucket_cap_mb: Noneddp_broadcast_buffers: Falsedataloader_pin_memory: Truedataloader_persistent_workers: Falseskip_memory_metrics: Trueuse_legacy_prediction_loop: Falsepush_to_hub: Falseresume_from_checkpoint: Nonehub_model_id: Nonehub_strategy: every_savehub_private_repo: Nonehub_always_push: Falsehub_revision: Nonegradient_checkpointing: Falsegradient_checkpointing_kwargs: Noneinclude_inputs_for_metrics: Falseinclude_for_metrics: []eval_do_concat_batches: Truefp16_backend: autopush_to_hub_model_id: Nonepush_to_hub_organization: Nonemp_parameters: auto_find_batch_size: Falsefull_determinism: Falsetorchdynamo: Noneray_scope: lastddp_timeout: 1800torch_compile: Falsetorch_compile_backend: Nonetorch_compile_mode: Noneinclude_tokens_per_second: Falseinclude_num_input_tokens_seen: noneftune_noise_alpha: Noneoptim_target_modules: Nonebatch_eval_metrics: Falseeval_on_start: Falseuse_liger_kernel: Falseliger_kernel_config: Noneeval_use_gather_object: Falseaverage_tokens_across_devices: Trueprompts: Nonebatch_sampler: batch_samplermulti_dataset_batch_sampler: round_robinrouter_mapping: {}learning_rate_mapping: {}| Epoch | Step | Training Loss |
|---|---|---|
| 0.3446 | 500 | 0.0194 |
| 0.6892 | 1000 | 0.0846 |
| 1.0338 | 1500 | 0.0424 |
| 1.3784 | 2000 | 0.0198 |
| 1.7229 | 2500 | 0.0171 |
| 2.0675 | 3000 | 0.0169 |
| 2.4121 | 3500 | 0.0039 |
| 2.7567 | 4000 | 0.003 |
@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",
}
@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}
}
Base model
google/embeddinggemma-300m