--- tags: - sentence-transformers - sentence-similarity - feature-extraction - generated_from_trainer - dataset_size:11397 - loss:MultipleNegativesRankingLoss base_model: sentence-transformers/all-mpnet-base-v2 widget: - source_sentence: men's sleeveless vest with a polished exterior and a tailored fit.. men's sleeveless vest with a polished exterior and a tailored fit. sentences: - 'Title: Arnodefrance Lity Of Gog Denim Jacket Graphic Print Washed Jacket Hip Pop Button Down Trucker Jacket Descripion: [''Arnodefrance provides more trendy clothing choices for trendy brand lovers and fashion icons. It has always been aimed at creating an international first-line trendy brand. It has unique cutting treatment, personalized color matching and comfortable soft fabrics. It expresses modern youth through clothing design. In a happy world, people play a self-style, create topics, and always maintain a trendy attitude to question common sense and pursue their own answers.'']' - 'Title: Columbia Girls'' Big Benton Fleece Jacket, Spring Blue/Blue Chill, Medium Descripion: ["There''s nothing more necessary than a fleece layer in a litter adventurer''s outdoor winter wardrobe—that''s why the Benton Springs Full Zip Fleece Jacket exists. Columbia''s soft, winter-ready jacket is the ultimate warmth provider and the everyday style piece. Crafted of our super-soft 100% polyester MTR filament fleece, this Benton Springs Full Zip Fleece Jacket is the perfect layering piece and first line of defense to combat the cold. It contains a modern classic fit that allows for comfortable movement and zippered side pockets to keep your small items (including your hands) secure. An added bonus is the warm collar that''s flexible so you can choose whether you want to wear it up or down, depending on your desired level of toastiness. Our Benton Springs Full Zip Fleece Jacket is available in many accommodating sizes and colors as well. To ensure the size you choose is right, utilize our sizing chart and the following measurement instructions: For the sleeves, start at the center back of your neck and measure across the shoulder and down to the sleeve. If you come up with a partial number, round up to the next even number. For the chest, measure at the fullest part of the chest, under the armpits and over the shoulder blades, keeping the tape measure firm and level."]' - 'Title: Men''s Slim Vest Sleeveless Jacket Casual PU Leather Vests Button Open V-Neck Simple Joker Slim Fit Vest Winter Descripion: [''SPECIFICATIONGender:MENFabric Type:BroadclothStyle:Smart CasualMaterial:NylonMaterial:ViscoseItem Type:Vests'']' - source_sentence: women's blazer with a tailored design, long sleeves, and a single-button closure. sentences: - "Title: Blazer Jackets for Women Lapel Long Sleeve Single Breasted Office Outerwear\ \ Solid Casual Long Coats Work Cardigans Descripion: ['☆☆☆☆☆☆▅▅▅▅▅▅▅▅▅▅' '☆☆☆☆☆☆▅▅▅▅▅▅▅▅▅▅'\ \ '☆☆☆☆☆☆▅▅▅▅▅▅▅▅▅▅'\n '▅▅▅▅▅▅▅▅▅▅▅▅▅▅▅▅' '▅▅▅▅▅▅▅▅▅▅▅▅▅▅▅▅' '▅▅▅▅▅▅▅▅▅▅▅▅▅▅▅▅'\ \ 'Q&A'\n 'Q1：Are these Anjikang store clothes true to size?A1：Yes, just order\ \ your size,we are standard US size.'\n \"Q2: Will it wrinkle or shrink after\ \ washed? or does it smell bad or itchy?A2: Not at all. We made of good material,so\ \ it won't had bad smelling,shrink or wrinkle,itchy all the problem you are worried.\"\ \n 'Q3: Does it look exactly like the picture?A3: Yes the color is the same as\ \ in the picture.'\n 'Q4: Washing instructions?A4: Hand wash recommended; Machine\ \ wash cold.'\n 'Q5: Does this material fade fast?A5: Not at all.'\n 'Q6: Is this\ \ soft or a rougher material?A6: Very soft and comfortable.'\n \"zip up jacket\ \ women womens dress coat long coats for women velvet coat winter sweaters shacket\ \ jacket women black varsity jacket bomber jackets flannel jackets rain jacket\ \ women varsity jacket women jean jacket with fringe jean jacket women hooded\ \ jean jacket velvet jackets water resistant jacket women cropped zip up jacket\ \ fleece tights for women winter womens plaid jacket denim long denim jackets\ \ plus size faux leather jacket women coats and jackets trench coats for women\ \ yellow plaid jacket women's windbreaker jackets camouflage jacket for women\ \ woman puffer jacket scrub jacket for women for nurses white suit jacket womens\ \ white cropped slim athletic yoga workout track sports zip up jacket flannel\ \ jacket white bomber jacket womens aztec denim jacket black parade jacket black\ \ winter coat women womens down jacket fall jacket women lightweight puffer jacket\ \ women black sequin jacket cape coat long plaid jacket women purple suit jacket\ \ white jackets for women\"]" - 'Title: Mother of The Bride Dresses with Jacket Lace Wedding Guest Dresses for Women Maxi Long Formal Evening Dress Descripion: [''Mother of the bride dresess chiffon evening dress formal evening party dresses a line mother of the bride dress'']' - "Title: 5665 Teen Girls Cape Coat for Women Long Plus Size Winter Warm Coat Button\ \ Thick Wool Peacoat Black Fleece Jacket S-5XL Descripion: ['♔Welcome to Our Store♔'\n\ \ '(◕ˇ∀ˇ◕) Have a nice shopping time, thank you so much! (›´ω`‹ )'\n '-----Size\ \ Note-----'\n 'Runs Small. We suggest buy one or two size larger. Thank you.'\n\ \ 'Please check the Size Chart before order. If you are not sure the size, please\ \ send message to us.Have a nice day!'\n \"Size.: Small US: 4 UK: 8 EU: 34 Bust:\ \ 101cm/39.76'' Shoulder: 67.5cm/26.57'' Sleeve: 41.5cm/16.34'' Length: 75cm/29.53''\"\ \n \"Size.: Medium US: 6 UK: 10 EU: 36 Bust: 106cm/41.73'' Shoulder: 70cm/27.56''\ \ Sleeve: 42cm/16.54'' Length: 76cm/29.92''\"\n \"Size.: Large US: 8 UK: 12 EU:\ \ 38 Bust: 111cm/43.70'' Shoulder: 72.5cm/28.54'' Sleeve: 42.5cm/16.73'' Length:\ \ 77cm/30.31''\"\n \"Size.: X-Large US: 10 UK: 14 EU: 40 Bust: 116cm/45.67'' Shoulder:\ \ 75cm/29.53'' Sleeve: 43cm/16.93'' Length: 78cm/30.71''\"\n \"Size.: XX-Large\ \ US: 12 UK: 16 EU: 42 Bust: 121cm/47.64'' Shoulder: 77.5cm/30.51'' Sleeve: 43.5cm/17.13''\ \ Length: 79cm/31.10''\"\n \"Size.: XXX-Large US: 14 UK: 18 EU: 44 Bust: 126cm/49.61''\ \ Shoulder: 80cm/31.50'' Sleeve: 44cm/17.32'' Length: 80cm/31.50''\"\n \"Size.:\ \ XXXX-Large US: 16 UK: 20 EU: 46 Bust: 131cm/51.57'' Shoulder: 82.5cm/32.48''\ \ Sleeve: 44.5cm/17.52'' Length: 81cm/31.89''\"\n \"Size.: XXXXX-Large US: 18\ \ UK: 21 EU: 48 Bust: 136cm/53.54'' Shoulder: 85cm/33.46'' Sleeve: 45cm/17.72''\ \ Length: 82cm/32.28''\"]" - source_sentence: men's tracksuit set featuring a hood, zip closure, and a comfortable fit with breathable fabric. sentences: - 'Title: INTL d.e.t.a.i.l.s Women''s Plus Size Packable Anorak Jacket Descripion: [''This plus size packable anorak jacket from Details is the perfect addition to your outerwear wardrobe. This is great for transitional seasons or collar spring/summer days or nights.'']' - "Title: Men's Linen Suits 2 Pieces Slim Fit Prom Suit Summer Beach Wedding Groomsman\ \ Jacket Pants Set Descripion: [\"Men's 2 Pieces Linen Suit Slim Fit Casual Summer\ \ Beach Suits for Men Formal Wedding Prom Business Tuxedo\"\n '● This suit contain\ \ 1 blazer, 1 pants● Selected High-quality Fabrics: Cotton, Polyester, Viscose.\ \ Selected Comfortable, Soft, Breathable Fabrics● Style: Classic Design, Slim\ \ Fit● Multi-Colors Optional: Provide Customized Colors'\n 'Slim Fit 3D Cut Blazer\ \ with Full Shoulder Design:'\n '2 Buttons Closure, Notch Lapel, 4 Pockets on\ \ front, 1 Vent'\n 'Strong and Durable Pants with Adjustable Waist:'\n 'Flat Front,\ \ Adjustable Waist Band' 'IMPORTANT TIPS About Size'\n '● Our Size: XS≈34R, S≈36R,\ \ M≈38R, L≈40R, XL≈42R, XXL≈44R, 3XL≈46R(The size analogy is for reference only,\ \ please check our size chart for the actual size)● PLEASE NOT look at the Amazon\ \ size chart. Please select \" customized color and size\" option if you need\ \ customized, then send us all measurements that listed below (The measurement\ \ guide pictures are in the left picture of the customization option)● Customized\ \ Size (units CM orinches): 1.Neckline 2.Shoulder to shoulder 3.Arm Length 4.Bicep\ \ 5.Cuff 6.Chest 7.Belly 8.Waist 9.Hips 10.Blazer Length 11.Pants Length 12.Thigh\ \ 13.Height 14.Weight=_kg or pounds● Please make sure all body measurements are\ \ correct, please feel free to contact us if you need help'\n 'Easy to Match with\ \ and Suitable for a lot of Occasions:'\n '● You can match with shirt, tie. You\ \ can also match with a solid color T-shirt, simple and comfortable● Suitable\ \ for Wedding, Business, Party, Many other occasions, also a great gift for someone\ \ important'\n 'With the cut somewhat narrow at the waist and legs, looks trendy,\ \ don’t be worry about the fit of the suit, you can enjoy the freedom of movement\ \ at the same time. Make yourself a modern and trimmed-down silhouette with this\ \ suit set, it will bring you tons of compliments!']" - 'Title: JG JENNY GHOO Men''s Casual Tracksuits Long Sleeve Jogging Suits Sweatsuit Sets Track Jackets and Pants 2 Piece Outfit Descripion: ["men''s tracksuits track suits for men hip hop sweatsuits jogging suits sets 2 piece Warm and breathable material. Great for everyday wear and for sport. This tracksuit has a soft and breathable material and it is suitable for any occasion. It has a hood and zippers and it is available in different colors and patterns."]' - source_sentence: a lightweight jacket for casual wear sentences: - 'Title: Umbro Brentford FC Mens 22/23 Presentation Jacket (L) (Black/Carbon) Descripion: [''Fabric: French Terry, Stretch, Woven. Design: Crest, Logo. Angular Panels, Branded Zip Pull, Inner Zip Guard, Side Panels. Fabric Technology: Lightweight. Sleeve-Type: Long-Sleeved. Neckline: Standing Collar. Pockets: 2 Side Pockets, Concealed Zip. Fastening: Full Zip. Hem: Clean Cut. 100% Officially Licensed.'']' - 'Title: AKNHD Baby Boys Girls Hooded Thick Snowsuit Romper Warm Snowsuit Coat Outwear Jacket Snowsuit with Gloves Descripion: ["Product Description:Fashion design,100% Brand New,high quality!Material: PolyesterPattern Type: SolidSleeve length: Long SleeveMain Color: As The Picture ShowStyle: FashionStylish and fashion design make your baby more attractiveGreat for casual, Daily, party or photoshoot, also a great idea for a baby show giftsIt is made of high quality materials,Soft hand feeling, no any harm to your baby''s skinPlease allow slight 1-3cm difference due to manual measurement and a little color variation for different display setting,thanks for your understanding!1 inch = 2.54 cmThank you and nice day!Package include:1PC Romper+1Pair Gloves/1PC Romper"]' - 'Title: Obermeyer Girls'' Katelyn Jacket Without FA Descripion: ["Our newly styled Katelyn is a luxurious jacket; Technical, sophisticated, and dependable for any endeavor. Children have no filters, They say what''s on their minds.They simply go and explore how things are, how they work, what they do. We love and encourage them to navigate their surroundings. Winter brings an excitement that opens her curiosity; The uniqueness of snowflakes entices us all, and for her, discovery and adventure."]' - source_sentence: enamel pin with a compact size, durable material, and a secure backing.. enamel pin with a compact size, durable material, and a secure backing. sentences: - "Title: Bleaches Kurosak Ichig Cosplay Hoodie Unisex Sweatshirt Jacket Pullover\ \ Urahar Kisuke Sweater Coat Streetwear HoodySweatshirt Hoody (X-Large, F-yellow\ \ 1) Descripion: ['Design：'\n 'Bleaches cosplay hoodie unisex Kurosak Ichig jacket\ \ pullover fashion hoody Urahar Kisuke long sleeve sweater coat adult Bleaches\ \ Kurosak Ichig sweatshirt hoodie tracksuit outerwear oversize girls boys.'\n\ \ \"Fabric: Made of high-quality polyester cotton, soft and comfortable fabric,\ \ suitable for men's daily wear. Material: Polyester. Hooded: With hat. Sleeve\ \ Length: Full sleeve, Long sleeve. Thickness: Standard. Season: Autumn Winter\ \ Spring. Style: Fashion, Creative, Funny, Casual, Hip Hop. Item Type: 2D printed\ \ hoodies sweatshirts adult. Pattern Type: Vivid 2D Print, Fashion Pattern 2D\ \ Printing. Package Includes: 1 X Anime Hoodie.\"]" - 'Title: Cute Cat Enamel Pin I LOVE ALL THE CATS Brooch Cartoon Animal Lapel Badge for Backpacks Jackets Clothes Bag Party Decoration Jewelry Gift for Friends Descripion: [''Cute Cat Enamel Pin I LOVE ALL THE CATS Brooch Cartoon Animal Lapel Badge for Backpacks Jackets Clothes Bag Party Decoration Jewelry Gift for Friends Cute Cat Enamel Pin I LOVE ALL THE CATS Brooch Cartoon Animal Lapel Badge for Backpacks Jackets Clothes Bag Party Decoration Jewelry Gift for Friends Cute Cat Enamel Pin I LOVE ALL THE CATS Brooch Cartoon Animal Lapel Badge for Backpacks Jackets Clothes Bag Party Decoration Jewelry Gift for Friends Cute Cat Enamel Pin I LOVE ALL THE CATS Brooch Cartoon Animal Lapel Badge for Backpacks Jackets Clothes Bag Party Decoration Jewelry Gift for Friends'']' - 'Title: Funny Chill Demon Enamel Pin Novelty Brooch Buttons Jewelry for Jackets Jeans Backpack Cloth Lapel Bag Hat Gift for Luci Fans Disenchantment Lovers Men Women Boy Girl Descripion: [''-Size - About 1.2" -Hard enamel -Black shiny metal -One rubber clutch'']' pipeline_tag: sentence-similarity library_name: sentence-transformers --- # SentenceTransformer based on sentence-transformers/all-mpnet-base-v2 This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [sentence-transformers/all-mpnet-base-v2](https://huggingface.co/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](https://huggingface.co/sentence-transformers/all-mpnet-base-v2) - **Maximum Sequence Length:** 128 tokens - **Output Dimensionality:** 768 tokens - **Similarity Function:** Cosine Similarity ### Model Sources - **Documentation:** [Sentence Transformers Documentation](https://sbert.net) - **Repository:** [Sentence Transformers on GitHub](https://github.com/UKPLab/sentence-transformers) - **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers) ### Full Model Architecture ``` SentenceTransformer( (0): Transformer({'max_seq_length': 128, '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}) ) ``` ## Usage ### Direct Usage (Sentence Transformers) First install the Sentence Transformers library: ```bash pip install -U sentence-transformers ``` Then you can load this model and run inference. ```python from sentence_transformers import SentenceTransformer # Download from the 🤗 Hub model = SentenceTransformer("knguyennguyen/mpnet_jacket4k_enhanced") # Run inference sentences = [ 'enamel pin with a compact size, durable material, and a secure backing.. enamel pin with a compact size, durable material, and a secure backing.', 'Title: Funny Chill Demon Enamel Pin Novelty Brooch Buttons Jewelry for Jackets Jeans Backpack Cloth Lapel Bag Hat Gift for Luci Fans Disenchantment Lovers Men Women Boy Girl Descripion: [\'-Size - About 1.2" -Hard enamel -Black shiny metal -One rubber clutch\']', "Title: Cute Cat Enamel Pin I LOVE ALL THE CATS Brooch Cartoon Animal Lapel Badge for Backpacks Jackets Clothes Bag Party Decoration Jewelry Gift for Friends Descripion: ['Cute Cat Enamel Pin I LOVE ALL THE CATS Brooch Cartoon Animal Lapel Badge for Backpacks Jackets Clothes Bag Party Decoration Jewelry Gift for Friends Cute Cat Enamel Pin I LOVE ALL THE CATS Brooch Cartoon Animal Lapel Badge for Backpacks Jackets Clothes Bag Party Decoration Jewelry Gift for Friends Cute Cat Enamel Pin I LOVE ALL THE CATS Brooch Cartoon Animal Lapel Badge for Backpacks Jackets Clothes Bag Party Decoration Jewelry Gift for Friends Cute Cat Enamel Pin I LOVE ALL THE CATS Brooch Cartoon Animal Lapel Badge for Backpacks Jackets Clothes Bag Party Decoration Jewelry Gift for Friends']", ] 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 #### Unnamed Dataset * Size: 11,397 training samples * Columns: sentence_0 and sentence_1 * Approximate statistics based on the first 1000 samples: | | sentence_0 | sentence_1 | |:--------|:----------------------------------------------------------------------------------|:-------------------------------------------------------------------------------------| | type | string | string | | details | | | * Samples: | sentence_0 | sentence_1 | |:----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------| | cosplay jacket designed for men, made from synthetic material, featuring a closure mechanism and suitable for various festive occasions. | Title: Poetic Walk Kill la Kill Cosplay Matoi Ryuko Costume Jacket Baseball Coat Uniform Sports Coat Descripion: ["Anime Kill la Kill Cosplay Matoi Ryuko Costume Jacket Baseball Coat Uniform Sports Coat Package:One good quality jacket. Fabric:Polyester. Size:Mens size,please choose size from size table,if you couldn't ensure the size,please email us your measurements:female/male,height,bust,waist and hip,then we could check which size fit for you . Occasion: Halloween,Birthday, Masquerade, Christmas, Carnival,theme parties,clothing parties, costume ball, family gatherings, Halloween Party .Cosplay and all kinds of seasonal holidays and parties ."] | | a collarless leather jacket for stylish outerwear | Title: Cole Haan Women's Leather Collarless Jacket Descripion: ['Collarless smooth lamb leather jacket with exposed snap detail at necline.'] | | jacket featuring a flexible closure, adjustable head covering, and secure storage options.. jacket featuring a flexible closure, adjustable head covering, and secure storage options. | Title: PUMA Puma X Helly Hansen Jacket Descripion: ['Equip Your Wardrobe With The Latest Styles And Technology From This Duo Of Sportswear Titans, Puma And Helly Hansen. Known For Their Excellence With Outerwear, Puma Has Teamed Up With The Experts Over At Helly Hansen To Produce High-Performance, High Style Options For This Line Of Winterwear.'] | * Loss: [MultipleNegativesRankingLoss](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#multiplenegativesrankingloss) with these parameters: ```json { "scale": 20.0, "similarity_fct": "cos_sim" } ``` ### Training Hyperparameters #### Non-Default Hyperparameters - `per_device_train_batch_size`: 128 - `per_device_eval_batch_size`: 128 - `num_train_epochs`: 5 - `multi_dataset_batch_sampler`: round_robin #### All Hyperparameters
Click to expand - `overwrite_output_dir`: False - `do_predict`: False - `eval_strategy`: no - `prediction_loss_only`: True - `per_device_train_batch_size`: 128 - `per_device_eval_batch_size`: 128 - `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-05 - `weight_decay`: 0.0 - `adam_beta1`: 0.9 - `adam_beta2`: 0.999 - `adam_epsilon`: 1e-08 - `max_grad_norm`: 1 - `num_train_epochs`: 5 - `max_steps`: -1 - `lr_scheduler_type`: linear - `lr_scheduler_kwargs`: {} - `warmup_ratio`: 0.0 - `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} - `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`: False - `hub_always_push`: False - `gradient_checkpointing`: False - `gradient_checkpointing_kwargs`: None - `include_inputs_for_metrics`: False - `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 - `batch_sampler`: batch_sampler - `multi_dataset_batch_sampler`: round_robin
### Framework Versions - Python: 3.11.11 - Sentence Transformers: 3.1.1 - Transformers: 4.45.2 - PyTorch: 2.5.1+cu121 - Accelerate: 1.2.1 - Datasets: 3.2.0 - Tokenizers: 0.20.3 ## Citation ### BibTeX #### Sentence Transformers ```bibtex @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 ```bibtex @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} } ```