您可以将经过SageMaker训练的模型加载到Keras中吗?

时间:2020-01-26 18:02:24

标签: keras object-detection amazon-sagemaker

我已遵循AWS教程(https://github.com/awslabs/amazon-sagemaker-examples/blob/master/introduction_to_amazon_algorithms/object_detection_pascalvoc_coco/object_detection_image_json_format.ipynb)并使用SageMaker训练了我的第一个模型。

最终结果是包含以下文件的档案: -hyperparams.json -model_algo_1-0000.params -model_algo_1-symbol.json

我不熟悉这种格式,因此无法通过keras.models.model_from_json()将其加载到Keras中

我假设这是另一种格式或AWS专有格式。

可以帮我确定格式吗? 是否可以将其加载到Keras模型中并在没有EC2实例(本地)的情况下进行推断?

谢谢!

1 个答案:

答案 0 :(得分:2)

内置算法是使用Apache MXNet实现的,因此您可以在本地加载模型。 load_checkpoint()是合适的API:https://mxnet.apache.org/api/python/docs/api/mxnet/model/index.html#mxnet.model.load_checkpoint