tensorflow 在还原的对象中找不到检查点中的值:(根目录).optimizer.iter

fkaflof6  于 2022-11-25  发布在  其他
关注(0)|答案(3)|浏览(519)

因此,我使用的是来自此链接的预先训练的权重:http://download.tensorflow.org/models/object_detection/tf2/20200711/ssd_mobilenet_v2_fpnlite_320x320_coco17_tpu-8.tar.gz然后我从下载链接将pipeline.config复制到我的文件夹,因为我想将优化器更改为adam,以便在我自己的数据集上训练(链接中的管道正在使用momentum_optimizer)
下面是我的pipeline.config要训练的代码:

optimizer {
adam_optimizer {
  learning_rate {
    cosine_decay_learning_rate {
      learning_rate_base: 0.01
      total_steps: 50000
      warmup_learning_rate: 0.026666
      warmup_steps: 1000
    }
  }
  epsilon: 1e-8
}
use_moving_average: false

}

但命令提示符告诉我

WARNING:tensorflow:Value in checkpoint could not be found in the restored object: (root).optimizer.iter
W0419 23:47:07.776149 17436 util.py:194] Value in checkpoint could not be found in the restored object: (root).optimizer.iter
WARNING:tensorflow:Value in checkpoint could not be found in the restored object: (root).optimizer.decay
W0419 23:47:07.777309 17436 util.py:194] Value in checkpoint could not be found in the restored object: (root).optimizer.decay
WARNING:tensorflow:Value in checkpoint could not be found in the restored object: (root).optimizer.momentum
W0419 23:47:07.779311 17436 util.py:194] Value in checkpoint could not be found in the restored object: (root).optimizer.momentum

有谁能解释一下请谢谢[1]:https://i.stack.imgur.com/BBmVA.png

1dkrff03

1dkrff031#

在执行以下代码行时,我也遇到了同样的错误:

python Tensorflow\models\research\object_detection\model_main_tf2.py --model_dir=Tensorflow\workspace\models\my_ssd_mobnet --pipeline_config_path=Tensorflow\workspace\models\my_ssd_mobnet\pipeline.config --num_train_steps=2000

改变步数对我起作用了!

pbossiut

pbossiut2#

对我有效的方法是进入模型检查点文件夹并删除所有以前的检查点,只留下管道配置文件。

vxf3dgd4

vxf3dgd43#

对我有效的方法是进入模型检查点文件夹并删除所有以前的检查点,只留下管道配置文件。
在我的操作功能中,需要在my_ssd_mobnet的地毯上删除检查点,并单独删除归档管线。配置并修改命令:python Tensorflow\模型\研究\对象_检测\模型_主_tf2.py--模型_目录=Tensorflow\工作空间\模型\我的_ssd_mobnet --管道_配置_路径=Tensorflow\工作空间\模型\我的_ssd_mobnet\管道.配置--训练_步数=2000

相关问题