Adattamento dello script di addestramento per registrare un hook

Debugger Amazon SageMaker è dotato di una libreria client denominata sagemaker-debugger Python SDK. sagemaker-debugger Python SDK fornisce strumenti per adattare lo script di addestramento prima dell'addestramento e strumenti di analisi dopo l'addestramento. In questa pagina, scoprirai come adattare lo script di addestramento utilizzando la libreria client.

sagemaker-debugger Python SDK fornisce funzioni wrapper che aiutano a registrare un hook per estrarre i tensori del modello, senza alterare lo script di addestramento. Per iniziare a raccogliere i tensori di output del modello ed eseguirne il debug per individuare problemi di addestramento, apporta le seguenti modifiche allo script di addestramento.