Tensor-Sammlungen mithilfe der CollectionConfig API konfigurieren

Verwenden Sie den CollectionConfig API-Vorgang, um Tensorsammlungen zu konfigurieren. Der Debugger bietet vorgefertigte Tensorsammlungen, die eine Vielzahl regulärer Ausdrücke (Regex) von Parametern abdecken, wenn Debugger-supported Deep-Learning-Frameworks und Algorithmen für maschinelles Lernen verwendet werden. Fügen Sie, wie im folgenden Beispielcode gezeigt, die integrierten Tensorsammlungen hinzu, die Sie debuggen möchten.

from sagemaker.core.debugger import CollectionConfig

collection_configs=[
    CollectionConfig(name="weights"),
    CollectionConfig(name="gradients")
]

Die vorherigen Sammlungen haben den Debugger-Hook so eingerichtet, dass er die Tensoren alle 500 Schritte basierend auf dem "save_interval" Standardwert speichert.

Eine vollständige Liste der verfügbaren integrierten Debugger-Sammlungen finden Sie unter Debugger-Sammlungen. Built-in

Wenn Sie die integrierten Sammlungen anpassen möchten, z. B. die Speicherintervalle und den Tensor-Regex ändern möchten, verwenden Sie die folgende CollectionConfig Vorlage, um die Parameter anzupassen.

from sagemaker.core.debugger import CollectionConfig

collection_configs=[
    CollectionConfig(
        name="tensor_collection",
        parameters={
            "key_1": "value_1",
            "key_2": "value_2",
            ...
            "key_n": "value_n"
        }
    )
]

Weitere Informationen zu verfügbaren Parameterschlüsseln finden Sie CollectionConfig im Amazon SageMaker Python SDK. Das folgende Codebeispiel zeigt beispielsweise, wie Sie die Speicherintervalle der Tensorsammlung „Verluste“ in verschiedenen Trainingsphasen anpassen können: Speicherverlust alle 100 Schritte in der Trainingsphase und Validierungsverlust alle 10 Schritte in der Validierungsphase.

from sagemaker.core.debugger import CollectionConfig

collection_configs=[
    CollectionConfig(
        name="losses",
        parameters={
            "train.save_interval": "100",
            "eval.save_interval": "10"
        }
    )
]