Yolo Inc

可能需要根据数据集的大小和通道数改一下相应的参数，预处理的过程

tf.train.Saver saves and restores variables. Please read the relevant snippet in the code. Hope it helps you.

可以参照 训练代码里的 evaluate_step 来写 test_model，当时用 python 训练，用别的语言载入模型进行预测，就没写 python 的 test_model。

抱歉刚看到，不知道你的问题是否解决了 我没遇到过类似问题，代码也在多个机器上跑过，下面是我觉得可能引起问题的原因： - 检查一下运行代码的环境是否有 GPU，如果有安装 nvida 驱动了吗 - 运行 lstm.sh 时 -CUDA_VISIBLE_DEVICES 参数是否与你机器中想用的 GPU 一致 希望对你有所帮助