RuntimeError: Only tuples, lists and Variables are supported as JIT inputs/outputs. Dictionaries and strings are also accepted, but their usage is not recommended. Here, received an input of unsupported type: numpy.ndarray这个问题的原因是什么?
时间: 2023-06-14 19:04:15 浏览: 576
Python RuntimeError: thread.__init__() not called解决方法
这个问题的原因是 PyTorch JIT(Just-In-Time)不支持 numpy.ndarray 类型的输入/输出。PyTorch JIT是用于加速 PyTorch 模型的运行的工具,它将模型编译成优化的代码以提高性能。在 JIT 中,只有元组、列表和变量是被支持的。因此,如果你尝试将一个 numpy.ndarray 传递给 JIT,你会收到这个错误。解决这个问题的方法是先将 numpy.ndarray 转换为 PyTorch 张量(torch.Tensor),然后再将其传递给 JIT。你可以使用 `torch.from_numpy` 函数将 numpy 数组转换为 PyTorch 张量。
阅读全文