我正在尝试在 algorithmia(一个 MLOps 平台)上部署 Pytorch 模型并返回一个 json 字符串作为输出。我定义了一个函数,它使用模型预测给定输入图像的对象的掩码并返回预测分数数组。data = { 'pred': json.dumps(pred.tolist()) }
我使用然后将其转换为 json json_string = json.dumps(data)
。当我想返回data
或json_string
显示上述错误时。
我尝试在本地进行测试。当我打印json_string = json.dumps(data)
它时,它不会抛出错误,尽管它会打印这个巨大的控制台输出几秒钟并完成执行。我不确定如何修改此代码以进行部署。