tensorflow object_detection api ERROR:tensorflow:Model diverged with loss = NaN.

最新推荐文章于 2022-09-10 14:38:29 发布

原创最新推荐文章于 2022-09-10 14:38:29 发布 · 3.3k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow #object #detection #error #NaN

错误专栏收录该内容

6 篇文章

订阅专栏

当使用model_main.py进行迁移学习训练遇到Model diverged with loss=NaN.错误时，可通过替换为legacy/train.py来解决。此外，升级TensorFlow至1.11版本也可能解决问题，但需谨慎操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

进行迁移学习时，使用model_main.py进行训练，无脑报错#“ERROR:tensorflow:Model diverged with loss = NaN.”。后来上github一查，看了一些言论，应该是版本问题，可以替换为models\research\object_detection\legacy\train.py 来进行训练。
也就说将原来你使用的：
python model_main.py --model_dir=training/ --pipeline_config_path=training/ssd_inception_v2_coco.config --logtostderr
替换为：
python ./legacy/train.py --pipeline_config_path=pipeline_config/ssd_mobilenet_v2_coco.config --train_dir=training/ --logtostderr
注意关键字不同
ps.也有人说将tf升级到1.11版本就行，但是升级真的牵一发动全身。。需谨慎，而且我还没试过。。不敢妄言