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

進行遷移學習時,使用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版本就行,但是升級真的牽一髮動全身。。需謹慎,而且我還沒試過。。不敢妄言

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章