当前位置: 代码迷 >> 综合 >> tensorflow object_detection api ERROR:tensorflow:Model diverged with loss = NaN.
  详细解决方案

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

热度:70   发布时间:2023-10-27 03:04:26.0

进行迁移学习时,使用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版本就行,但是升级真的牵一发动全身。。需谨慎,而且我还没试过。。不敢妄言

  相关解决方案