部署Tensorflow模型，获取错误

Question

我将训练有素的Tensorflow模型导入android。我保存了我的输入和输出张量，即＆＃34;输入＆＃34;和＆＃34; y _＆＃34;分别。

这是我在Android Studio中的代码：

public float[] getActivityProb(float[] input_signal)
    {
        float[] result = new float[OUTPUT_SIZE];
        inferenceInterface.feed(INPUT_NODE,input_signal,INPUT_SIZE);
        inferenceInterface.run(OUTPUT_NODES);
        inferenceInterface.fetch(OUTPUT_NODE,result);
        return result;
    }

然后这个函数返回输出的概率：

inferenceInterface.run(OUTPUT_NODES);

但我的应用程序崩溃了。我确信输入和输出尺寸是正确的（或者可能不是）。但我检查它们是相同的（在Android Studio中声明的大小和Python中的实际大小）。

这个错误有可能是什么？有没有更好的方法在Android Studio中调试？

另外，我找到了导致错误的行：

os.environ['CUDA_VISIBLE_DEVICES'] = '1'