フックを登録するためにトレーニングスクリプトを適応させる

Amazon SageMaker Debugger には sagemaker-debugger Python SDK と呼ばれるクライアントライブラリが付属しています。sagemaker-debugger Python SDK には、トレーニング前にトレーニングスクリプトを調整するためのツールと、トレーニング後に分析ツールを提供します。このページでは、クライアントライブラリを使用してトレーニングスクリプトを適合させる方法について説明します。

sagemaker-debugger Python SDK には、トレーニングスクリプトの変更することなく、モデルテンソルを抽出するフックの登録に役立つラッパー関数が用意されています。モデル出力テンソルを収集を開始し、デバッグしてトレーニングの問題を見つけるには、トレーニングスクリプトに次の変更を加えます。