Un accumulatore condizionale per aggregare gradienti.
L'accumulatore accetta gradienti contrassegnati con local_step maggiore o uguale al più recente global_step noto all'accumulatore. La media può essere estratta dall'accumulatore, a condizione che siano stati accumulati gradienti sufficienti. L'estrazione della media reimposta automaticamente l'aggregato a 0 e incrementa il global_step registrato dall'accumulatore. Questa è una versione risorsa di ConditionalAccumulator che funzionerà in TF2.0 con tf.cond versione 2.
Classi annidate
classe | ResourceConditionalAccumulator.Options | Attributi facoltativi per ResourceConditionalAccumulator |
Metodi pubblici
Output <Object> | asOutput () Restituisce la maniglia simbolica di un tensore. |
statico ResourceConditionalAccumulator.Options | container (String container) |
static <T> ResourceConditionalAccumulator | create ( ambito ambito, classe <T> dtype, forma forma, opzioni ... opzioni) Metodo Factory per creare una classe che esegue il wrapping di una nuova operazione ResourceConditionalAccumulator. |
Uscita <?> | maniglia () La maniglia dell'accumulatore. |
statico ResourceConditionalAccumulator.Options | reductionType (String reductionType) |
statico ResourceConditionalAccumulator.Options | sharedName (stringa sharedName) |
Metodi ereditati
Metodi pubblici
output pubblico <Object> asOutput ()
Restituisce la maniglia simbolica di un tensore.
Gli input per le operazioni TensorFlow sono output di un'altra operazione TensorFlow. Questo metodo viene utilizzato per ottenere un handle simbolico che rappresenta il calcolo dell'input.
public static ResourceConditionalAccumulator.Options container (String container)
Parametri
contenitore | Se non vuoto, questo accumulatore viene posto nel contenitore dato. In caso contrario, viene utilizzato un contenitore predefinito. |
---|
public static ResourceConditionalAccumulator create ( ambito ambito, classe <T> dtype, forma forma, opzioni ... opzioni)
Metodo Factory per creare una classe che racchiude una nuova operazione ResourceConditionalAccumulator.
Parametri
scopo | ambito attuale |
---|---|
dtype | Il tipo di valore che viene accumulato. |
forma | La forma dei valori può essere [], nel qual caso la forma è sconosciuta. |
opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di ResourceConditionalAccumulator
public static ResourceConditionalAccumulator.Options sharedName (String sharedName)
Parametri
sharedName | Se non è vuoto, questo accumulatore verrà condiviso con il nome specificato in più sessioni. |
---|