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

fix data load error in static mode (#55541)

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