guillaume-be/rust-bert
1
1
Fork 0
mirror of https://github.com/guillaume-be/rust-bert.git synced 2024-09-11 12:55:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

Updated generation benchmark configuration

Browse Source
This commit is contained in:
Guillaume B 2021-01-24 11:13:50 +01:00
parent 56574a15d5
commit 822f0bf19f
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
benches/generation_benchmark.rs
View File

@ -33,6 +33,8 @@ fn create_text_generation_model() -> TextGenerationModel {
repetition_penalty: 1.0,
length_penalty: 1.0,
no_repeat_ngram_size: 3,
num_beam_groups: None,
diversity_penalty: None,
num_return_sequences: 5,
device: Device::cuda_if_available(),
};