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

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值