Skip to content

Commit

Permalink
Merge pull request #61 from sbintuitions/fix/readme_dpsbert
Browse files Browse the repository at this point in the history
[Fix] `DataParallelSentenceBertEmbedder` in README
  • Loading branch information
lsz05 authored Aug 30, 2024
2 parents e4eb746 + ad1f6d4 commit f4b272e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -57,12 +57,12 @@ poetry run python -m jmteb \

There are two ways to enable multi-GPU evaluation.

* New class `DPSentenceBertEmbedder` ([here](src/jmteb/embedders/data_parallel_sbert_embedder.py)).
* New class `DataParallelSentenceBertEmbedder` ([here](src/jmteb/embedders/data_parallel_sbert_embedder.py)).

```bash
poetry run python -m jmteb \
--evaluators "src/configs/tasks/jsts.jsonnet" \
--embedder DPSentenceBertEmbedder \
--embedder DataParallelSentenceBertEmbedder \
--embedder.model_name_or_path "<model_name_or_path>" \
--save_dir "output/<model_name_or_path>"
```
Expand Down

0 comments on commit f4b272e

Please sign in to comment.