keysentence-finder / training_arguments.json
m3hrdadfi's picture
Initial
eb3d3e8
{"model": "pritamdeka/S-PubMedBert-MS-MARCO", "dataset": "/mimer/NOBACKUP/groups/addcell/biomedical-data-extraction/sentence_transformers/data/prepared_data/Annotated_dataset/general_query/", "verbose": "2", "evaluator": "binary", "evaluation_steps": "1000", "batch_size": "32", "epochs": "1", "warmup": "100", "lr": "2e-05", "loss_function": "contrastive", "output_dir": "/mimer/NOBACKUP/groups/addcell/biomedical-data-extraction/sentence_transformers/data/models/general_query_all/", "token": "-1", "device_name": "NVIDIA A40", "training_time_seconds": 4652.426364183426}