Я запускаю export_inference_graph.py
, чтобы экспортировать граф вывода, он успешно выполняется и граф экспортируется в течение нескольких секунд. Но я недавно обучил модель Faster RCNN с очень большим набором данных, обучение для этого было сравнительно медленным, и папка, где находятся все контрольные точки, имеет размер примерно 56 ГБ. Я запускаю команду:
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 ГБ.