从Sagemaker Estimator训练后获得权重

Question

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

Answer 1

对于SageMaker Python SDK中的Estimator对象，调用fit()后，您可以调用来获取模型工件的S3 URL，

model_artifacts_url = estimator.create_model().model_data

模型本身作为压缩包保存在此位置的S3存储桶中。因此，您可以从S3中获取模型参数。