总的来说,我对 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 专业版
感谢所有帮助!