Fix data type error

cd14bbe0 · Steffy-zxf · GitHub · 3eb4d334 · cd14bbe0
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddlehub/datasets/base_nlp_dataset.py paddlehub/datasets/base_nlp_dataset.py +1 -1

未找到文件。
--- a/paddlehub/datasets/base_nlp_dataset.py
+++ b/paddlehub/datasets/base_nlp_dataset.py
@@ -276,7 +276,7 @@ class TextClassificationDataset(BaseNLPDataset, paddle.io.Dataset):
            if Version(paddlenlp.__version__) >= Version('2.0.0rc5'):
                token_type_ids = np.array(record['token_type_ids'])
            else:
-                token_type_ids = record['segment_ids']
+                token_type_ids = np.array(record['segment_ids'])

            if 'label' in record.keys():
                return input_ids, token_type_ids, np.array(record['label'], dtype=np.int64)