如何使用训练有素的模型预测 Cupy 数组？
Posted
技术标签:
【中文标题】如何使用训练有素的模型预测 Cupy 数组？【英文标题】：How to predict cupy array using a trained model? 【发布时间】：2021-07-28 03:32:56 【问题描述】：我想使用经过训练的模型进行预测，但该模型似乎无法将 Cupy 数据作为输入。我的代码的sn-ps如下：
import cupy as cp
model_evaluator = my_model()
model_evaluator.build_model()
model_evaluator.model.load_weights(trained_model)
model_evaluator.model.predict(my_cupy_data) # throws error here
model_evaluator.model.predict(cp.asnumpy(my_cupy_data)) # works fine
操作系统：Ubuntu 18.04
ENV：conda + python 3.6 + tensorflow 2.1.0
错误：InvalidArgumentError TypeError: len() of unsized object
【问题讨论】：
【参考方案1】：尝试将 CuPy 数组转换为 TensorFlow 张量。
tensor = tf.experimental.dlpack.from_dlpack(cupy_array.toDlpack())
【讨论】：
以上是关于如何使用训练有素的模型预测 Cupy 数组？的主要内容，如果未能解决你的问题，请参考以下文章