From c9e927227ce2cf1737ca6b33002bf1eebd05ec64 Mon Sep 17 00:00:00 2001 From: raccoonliukai <903896015@qq.com> Date: Tue, 13 Aug 2019 20:39:03 +0800 Subject: [PATCH] fix(encoder): use offline model in ci-base for pytorch transformer --- tests/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/__init__.py b/tests/__init__.py index 07f9da59..b47c1d0b 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -51,7 +51,7 @@ def line2pb_doc(line: str, doc_id: int = 0, deliminator: str = r'[.。!?!?]+ 'MOBILENET_MODEL': '/', 'FASTERRCNN_MODEL': '/', 'GNES_PROFILING': '', - 'TORCH_TRANSFORMERS_MODEL': 'bert-base-uncased', + 'TORCH_TRANSFORMERS_MODEL': '/torch_transformer' }, 'idc-165': { 'BERT_CI_PORT': 7125,