进行迁移学习时,使用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版本就行,但是升级真的牵一发动全身。。需谨慎,而且我还没试过。。不敢妄言
tensorflow object_detection api ERROR:tensorflow:Model diverged with loss = NaN.
最新推荐文章于 2022-09-10 14:38:29 发布