未验证 提交 746e7cdc 编写于 作者: K Kai Song 提交者: GitHub

fix data load error in static mode (#55541)

上级 44f409cf
......@@ -384,7 +384,7 @@ def _worker_loop(
tensor_list = [
numpy2lodtensor(b)
if isinstance(b, np.ndarray)
else b.value().get_tensor()
else b.get_tensor()
for b in batch
]
out_queue.put((idx, tensor_list, structure))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册