You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@singa.apache.org by jo...@apache.org on 2020/09/24 03:30:48 UTC
[singa] branch dev updated: fix some issues and requirements
This is an automated email from the ASF dual-hosted git repository.
joddiyzhang pushed a commit to branch dev
in repository https://gitbox.apache.org/repos/asf/singa.git
The following commit(s) were added to refs/heads/dev by this push:
new 76c2704 fix some issues and requirements
76c2704 is described below
commit 76c2704d5bfe1195603179f3635acb8338dba695
Author: joddiy <jo...@qq.com>
AuthorDate: Thu Sep 24 11:28:31 2020 +0800
fix some issues and requirements
---
examples/onnx/gpt2/gpt2.py | 2 +-
examples/onnx/ro_bert_a/requirements.txt | 1 +
examples/onnx/{ => ro_bert_a}/ro_bert_a.py | 12 ++----------
3 files changed, 4 insertions(+), 11 deletions(-)
diff --git a/examples/onnx/gpt2/gpt2.py b/examples/onnx/gpt2/gpt2.py
index 56b7cfe..dd27334 100644
--- a/examples/onnx/gpt2/gpt2.py
+++ b/examples/onnx/gpt2/gpt2.py
@@ -106,5 +106,5 @@ if __name__ == "__main__":
y = tensor.Tensor(device=dev, data=y)
x = tensor.concatenate([x, y], 2)
- text = tokenizer.decode(output)
+ text = postprocess(output)
print(text)
\ No newline at end of file
diff --git a/examples/onnx/ro_bert_a/requirements.txt b/examples/onnx/ro_bert_a/requirements.txt
new file mode 100644
index 0000000..14693ad
--- /dev/null
+++ b/examples/onnx/ro_bert_a/requirements.txt
@@ -0,0 +1 @@
+transformers==2.5.1
\ No newline at end of file
diff --git a/examples/onnx/ro_bert_a.py b/examples/onnx/ro_bert_a/ro_bert_a.py
similarity index 95%
rename from examples/onnx/ro_bert_a.py
rename to examples/onnx/ro_bert_a/ro_bert_a.py
index 5b6ac0a..b6b02ed 100644
--- a/examples/onnx/ro_bert_a.py
+++ b/examples/onnx/ro_bert_a/ro_bert_a.py
@@ -17,10 +17,6 @@
# under the License.
#
-
-import sys
-sys.path.append('/singa/build/python/')
-
import os
import numpy as np
@@ -30,6 +26,8 @@ from singa import sonnx
from singa import autograd
import onnx
+import sys
+sys.path.append(os.path.dirname(__file__) + '/..')
from utils import download_model, check_exist_or_download
import logging
@@ -45,12 +43,6 @@ def preprocess():
tokens = np.array(tokens)
return tokens.reshape([1, -1]).astype(np.float32)
-
-def postprocess(out):
- text = tokenizer.decode(out)
- return text
-
-
class MyModel(sonnx.SONNXModel):
def __init__(self, onnx_model):