提交 dee34721 编写于 作者: W wjj19950828

fixed bugs

上级 027fb18c
...@@ -14,7 +14,9 @@ ...@@ -14,7 +14,9 @@
from six import text_type as _text_type from six import text_type as _text_type
from x2paddle import program from x2paddle import program
from utils import ConverterCheck # import sys
# sys.path.append('.')
from x2paddle.utils import ConverterCheck
import argparse import argparse
import sys import sys
import logging import logging
......
...@@ -14,6 +14,7 @@ ...@@ -14,6 +14,7 @@
# limitations under the License. # limitations under the License.
import paddle import paddle
import x2paddle
import requests import requests
import threading import threading
...@@ -43,13 +44,12 @@ class ConverterCheck(threading.Thread): ...@@ -43,13 +44,12 @@ class ConverterCheck(threading.Thread):
def __init__(self, def __init__(self,
task="onnx", task="onnx",
version=__version__,
convert_state=None, convert_state=None,
lite_state=None, lite_state=None,
extra_info=None): extra_info=None):
threading.Thread.__init__(self) threading.Thread.__init__(self)
self._task = task self._task = task
self._version = version self._version = x2paddle.__version__
self._convert_state = convert_state self._convert_state = convert_state
self._lite_state = lite_state self._lite_state = lite_state
self._extra_info = extra_info self._extra_info = extra_info
...@@ -57,13 +57,13 @@ class ConverterCheck(threading.Thread): ...@@ -57,13 +57,13 @@ class ConverterCheck(threading.Thread):
def run(self): def run(self):
params = { params = {
'task': self._task, 'task': self._task,
'version': self._version, 'x2paddle_version': self._version,
'paddle_version': paddle.__version__, 'paddle_version': paddle.__version__,
'convert_state': self._convert_state, 'convert_state': self._convert_state,
'from': 'x2paddle' 'from': 'x2paddle'
} }
if self._lite_state is not None: if self._lite_state is not None:
params.update(self._lite_state) params.update({'lite_state': self._lite_state})
if self._extra_info is not None: if self._extra_info is not None:
params.update(self._extra_info) params.update(self._extra_info)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册