使用带有自定义参数值的调试器内置规则 - Amazon SageMaker
Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅 中国的 Amazon Web Services 服务入门 (PDF)

本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。

使用带有自定义参数值的调试器内置规则

如果您要调整内置的规则参数值并自定义张量集合正则表达式,请配置 ProfilerRule.sagemakerRule.sagemaker 类方法的 base_configrule_parameters 参数。使用 Rule.sagemaker 类方法时,您也可以通过 collections_to_save 参数自定义张量集合。使用配置张量集合 CollectionConfig API 中提供了如何使用 CollectionConfig 类的说明。

为内置规则使用以下配置模板来自定义参数值。通过根据需要更改规则参数,您可以调整规则触发的敏感度。

  • 您在 base_config 参数中调用内置规则方法。

  • rule_parameters 参数用于调整 调试器内置规则列表 中列出的内置规则的默认键值。

  • collections_to_save参数通过,它需要nameparameters参数 CollectionConfigAPI,采用张量配置。

有关调试器规则类、方法和参数的更多信息,请参阅 Amaz on Python SageMaker 中的SageMaker调试器规则类。SDK

from sagemaker.debugger import Rule, ProfilerRule, rule_configs, CollectionConfig rules=[ Rule.sagemaker( base_config=rule_configs.built_in_rule_name(), rule_parameters={ "key": "value" }, collections_to_save=[ CollectionConfig( name="tensor_collection_name", parameters={ "key": "value" } ) ] ) ]

各个规则的参数描述和值自定义示例均在 调试器内置规则列表 中提供。