amazon-web-services - 从 Sagemaker Estimator 训练后获取权重

Question

Tensorflow 的 Estimator 提供了一种在使用get_variable_value进行训练/测试后获取所需变量值的方法。Sagemaker 的 Estimator 中是否存在类似的功能，以便我能够在训练模型后获得权重。

标签： amazon-web-servicesamazon-sagemaker

Answer 1

对于EstimatorSageMaker Python SDK 中的对象，在您调用之后，fit()您可以调用 get the S3 URL of your model artifacts with

model_artifacts_url = estimator.create_model().model_data

模型本身作为 tarball 在此位置保存到您的 S3 存储桶中。所以从这里你可以从 S3 中获取模型参数。