Я пытаюсь получить модель для экспорта в формате, создаваемом tf.saved_model.save, который представляет собой большой файл .pb, содержащий всю модель и веса.
У меня есть
estimator.export_saved_model(
export_dir_base=os.environ.get('SM_MODEL_DIR'),
serving_input_receiver_fn=serving_input_fn
)
для обучающего сценария в режиме сценария Sagemaker, но, похоже, он выводит формат protobuf с большой папкой переменных и небольшим файлом .pb. Как мне экспортировать полную модель в формате .pb?
Это выполняется в форме сценария на обучающем экземпляре Sagemaker, поэтому я хотел бы, если возможно, избегать ручных команд freeze_graph.py, выполняемых локально.