SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix ROUGE add example check and update README (#18398)

Browse Source 
* Fix ROUGE add example check and update README

* Stay consistent in values
This commit is contained in:
Sylvain Gugger
2022-08-01 11:14:49 -04:00
committed by GitHub
parent 62098b9348
commit 941d233153
17 changed files with 60 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
examples/pytorch/multiple-choice/run_swag_no_trainer.py
View File

@@ -52,9 +52,12 @@ from transformers import (
default_data_collator,
get_scheduler,
)
from transformers.utils import PaddingStrategy, get_full_repo_name, send_example_telemetry
from transformers.utils import PaddingStrategy, check_min_version, get_full_repo_name, send_example_telemetry
# Will error if the minimal version of Transformers is not installed. Remove at your own risks.
check_min_version("4.22.0.dev0")
logger = get_logger(__name__)
# You should update this to your particular problem to have better documentation of `model_type`
MODEL_CONFIG_CLASSES = list(MODEL_MAPPING.keys())