Browse Source

修改api.load()函数

tags/v0.2.0
yh_cc 6 years ago
parent
commit
ae0cc9a46b
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      fastNLP/api/api.py

+ 2
- 2
fastNLP/api/api.py View File

@@ -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']

Loading…
Cancel
Save