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

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

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值