From ae0cc9a46bba7a5de0b0b5c4f9846ab74259e536 Mon Sep 17 00:00:00 2001
From: yh_cc <poemsmileyh@gmail.com>
Date: Sat, 10 Nov 2018 10:31:45 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9api.load()=E5=87=BD=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 fastNLP/api/api.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/fastNLP/api/api.py b/fastNLP/api/api.py
index b557038b..9c20c2a6 100644
--- a/fastNLP/api/api.py
+++ b/fastNLP/api/api.py
@@ -1,5 +1,5 @@
 
-import _pickle
+import torch
 
 
 class API:
@@ -11,6 +11,6 @@ class API:
         pass
 
     def load(self, name):
-        _dict = _pickle.load(name)
+        _dict = torch.load(name)
         self.pipeline = _dict['pipeline']
         self.model = _dict['model']