TF 2.4.1 使用预训练对象检测模型训练时挂起

时间:2021-02-22 18:57:02

标签: python tensorflow tensorflow2.0 object-detection-api

总的来说,我对 TF 和 ML 还很陌生,所以我非常依赖 TF 提供的文档和教程。我一直在关注这封信的 Tensorflow 2.0 Objection Detection API 教程,但在训练时遇到了一个问题:每次我运行训练脚本 model_main_tf2.py 时,它总是在输出后挂起:
I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:116] None of the MLIR optimization passes are enabled (registered 2) 在多次折旧警告之后。我尝试了许多不同的方法来解决这个问题,包括修改 train 脚本和 pipeline.config 文件。我的数据集不是很大,少于 100 张图片,每张图片最多 15 个标签。

有用信息:
Python 3.8.0
Tensorflow 2.4.4(非 GPU)
Windows 10 专业版

感谢所有帮助!

0 个答案:

没有答案