Carica i parametri di incorporamento di ADAM con il supporto del debug.
Un'operazione che carica i parametri di ottimizzazione in HBM per l'incorporamento. Deve essere preceduto da un'operazione ConfigureTPUEmbeddingHost che imposta la corretta configurazione della tabella di incorporamento. Ad esempio, questa operazione viene utilizzata per installare i parametri caricati da un checkpoint prima che venga eseguito un ciclo di addestramento.
Classi annidate
classe | LoadTPUEmbeddingADAMParametersGradAccumDebug.Options | Attributi facoltativi per LoadTPUEmbeddingADAMParametersGradAccumDebug |
Metodi pubblici
static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options | config (String config) |
static LoadTPUEmbeddingADAMParametersGradAccumDebug | create ( Scope scope, Operando <Float> parametri, Operando <Float> Momenta, Operando <Float> velocità, Operando <Float> Gradiente Accumulatori, Long numShards, Long shardId, Opzioni ... Metodo Factory per creare una classe che avvolge una nuova operazione LoadTPUEmbeddingADAMParametersGradAccumDebug. |
static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options | tableId (Long tableId) |
static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options | tableName (stringa tableName) |
Metodi ereditati
Metodi pubblici
public static LoadTPUEmbeddingADAMParametersGradAccumDebug create ( Scope scope, Operand <Float> parameters, Operand <Float> Momenta, Operand <Float> velocities, Operand <Float> gradientAccumulators, Long numShards, Long shardId, Options ...
Metodo Factory per creare una classe che avvolge una nuova operazione LoadTPUEmbeddingADAMParametersGradAccumDebug.
Parametri
scopo | ambito attuale |
---|---|
parametri | Valore dei parametri utilizzati nell'algoritmo di ottimizzazione di ADAM. |
Momenta | Valore di Momenta utilizzato nell'algoritmo di ottimizzazione di ADAM. |
velocità | Valore delle velocità utilizzate nell'algoritmo di ottimizzazione di ADAM. |
gradiente Accumulatori | Valore di gradiente_accumulatori utilizzati nell'algoritmo di ottimizzazione di ADAM. |
opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di LoadTPUEmbeddingADAMParametersGradAccumDebug