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

examples/seq2seq/__init__.py mutates sys.path (#7194)

Browse Source
This commit is contained in:
Stas Bekman
2020-09-20 13:54:42 -07:00
committed by GitHub
parent a4faeceaed
commit 7cbf0f722d
8 changed files with 58 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
examples/seq2seq/run_eval.py
View File

@@ -11,14 +11,11 @@ import torch
from tqdm import tqdm
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
from utils import calculate_bleu, calculate_rouge, parse_numeric_n_bool_cl_kwargs, use_task_specific_params
logger = getLogger(__name__)
try:
from .utils import calculate_bleu, calculate_rouge, parse_numeric_n_bool_cl_kwargs, use_task_specific_params
except ImportError:
from utils import calculate_bleu, calculate_rouge, parse_numeric_n_bool_cl_kwargs, use_task_specific_params
DEFAULT_DEVICE = "cuda" if torch.cuda.is_available() else "cpu"