要解决此错误,需要执行以下步骤:
使用SageMaker Studio或Jupyter Notebook创建一个新的notebook。
安装sagemaker-debugger库。可以使用以下命令在notebook中安装:
!pip install sagemaker-debugger
import sagemaker
sess = sagemaker.Session()
region = sess.boto_region_name
from sagemaker.debugger import ( DebuggerHookConfig, rule_configs, CollectionConfig, )
from sagemaker.estimator import Estimator
estimator = Estimator( image_uri="123456789012.dkr.ecr.us-east-2.amazonaws.com/my-custom-image", role="SageMakerRole", instance_count=1, instance_type="ml.c4.xlarge", sagemaker_session=sess, debugger_hook_config=DebuggerHookConfig( s3_output_path="s3://sagemaker/{}/{}/debug".format(region, sess.default_bucket()), collection_configs=[ CollectionConfig(name="all", parameters={"all": "true"}), ], hook_parameters={ "save_interval": "10", }, # TODO add debugger rules here. rules=[rule_configs.loss_not_decreasing(), rule_configs.overfit()], ), )
这样就能够成功地加载估计器的调试器信息。