Tensorflow-无法为检查点导出非常大的文件的推理图

时间:2019-06-26 19:25:03

标签: python tensorflow deep-learning computer-vision

我运行export_inference_graph.py来导出推理图,它成功运行并且该图在几秒钟内被导出。但是,我最近用一个非常大的数据集训练了Faster RCNN模型,对此的训练相对较慢,并且所有检查点都在其中的文件夹的大小约为56 GB。我运行命令:

python export_inference_graph.py \
    --input_type image_tensor \
    --pipeline_config_path fasterrcnn.config \
    --trained_checkpoint_prefix training/model.ckpt-41671 \
    --output_directory inference_graph

,它可以连续运行几个小时,什么也没有。文件model.ckpt-41671.data-00000-of-00001的大小为6.3 GB。

0 个答案:

没有答案