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

Cleanup

Browse Source 
Improve global visibility on the run_squad script, remove unused files and fixes related to XLNet.
This commit is contained in:
LysandreJik
2019-12-05 16:01:51 -05:00
parent 9ecd83dace
commit e9217da5ff
5 changed files with 45 additions and 1387 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
transformers/data/processors/squad.py
View File

@@ -9,7 +9,7 @@ from ...tokenization_bert import BasicTokenizer, whitespace_tokenize
from .utils import DataProcessor, InputExample, InputFeatures
from ...file_utils import is_tf_available, is_torch_available
if is_torch_available:
if is_torch_available():
import torch
from torch.utils.data import TensorDataset