在某个检查点冻结模型，而不是最后一个带有exporter_main_v2的模型

问题描述

我使用tf 2.2训练了模型，并使用model_main_tf2.py训练了对象检测API 在步骤2000之后，损失大大增加，这是我的输出中的ckpt 3 我想将模型冻结在ckpt 3而不是ckpt4

使用exporter_main_v2.py将模型冻结在我不需要的最新ckpt上

如何在ckpt 3冻结模型

我使用的导出器主命令

python ./object_detection/exporter_main_v2.py \
    --input_type image_tensor \
    --pipeline_config_path /home/mai/ssd_mobilenet_v2_320x320_coco17_tpu-8.config \
    --trained_checkpoint_dir /home/mai/ssd_ckpt/ssd_mobilenet_v2_320x320_coco17_tpu-8freeze_fext_2.1kb \
    --output_directory /home/mai/ssd_ckpt/frozen_models/ssd_mobilenet_v2_320x320_coco17_tpu-8freeze_fext_2.1kb

该文件夹中的文件：

$ ls 
checkpoint                  ckpt-2.index
ckpt-1.data-00000-of-00002  ckpt-3.data-00000-of-00002
ckpt-1.data-00001-of-00002  ckpt-3.data-00001-of-00002
ckpt-1.index                ckpt-3.index
ckpt-2.data-00000-of-00002  eval
ckpt-2.data-00001-of-00002  train

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）