未验证 提交 35b8b083 编写于 作者: X Xiaoyao Xi 提交者: GitHub

Merge pull request #30 from wangxiao1021/windows-fix

fix bugs
...@@ -161,14 +161,14 @@ def create_joint_iterator_fn(iterators, iterator_prefixes, joint_shape_and_dtype ...@@ -161,14 +161,14 @@ def create_joint_iterator_fn(iterators, iterator_prefixes, joint_shape_and_dtype
if 'token_ids' in outputs: if 'token_ids' in outputs:
val1 = len(outputs['token_ids']) val1 = len(outputs['token_ids'])
val = _check_and_adapt_shape_dtype([val1], [[1], 'int64']) val = _check_and_adapt_shape_dtype(np.array([val1], dtype='int64'), [[1], 'int64'], iterator_prefixes[id]+' tokenids: ')
results[outname_to_pos['batch_size']] = val results[outname_to_pos['batch_size']] = val
val2 = len(outputs['token_ids'][0]) val2 = len(outputs['token_ids'][0])
val = _check_and_adapt_shape_dtype([val2], [[1], 'int64']) val = _check_and_adapt_shape_dtype(np.array([val2], dtype='int64'), [[1], 'int64'])
results[outname_to_pos['seqlen']] = val results[outname_to_pos['seqlen']] = val
val = _check_and_adapt_shape_dtype([val1*val2], [[1], 'int64']) val = _check_and_adapt_shape_dtype(np.array([val1*val2], dtype='int64'), [[1], 'int64'])
results[outname_to_pos['batchsize_x_seqlen']] = val results[outname_to_pos['batchsize_x_seqlen']] = val
else: else:
if not has_show_warn: if not has_show_warn:
......
...@@ -21,7 +21,7 @@ Authors: zhouxiangyang(zhouxiangyang@baidu.com) ...@@ -21,7 +21,7 @@ Authors: zhouxiangyang(zhouxiangyang@baidu.com)
Date: 2019/09/29 21:00:01 Date: 2019/09/29 21:00:01
""" """
import setuptools import setuptools
with open("README.md", "r") as fh: with open("README.md", "r", encoding='utf-8') as fh:
long_description = fh.read() long_description = fh.read()
setuptools.setup( setuptools.setup(
name="paddlepalm", name="paddlepalm",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册