Searched defs:use_gradient_accumulation (Results 1 – 2 of 2) sorted by relevance
155 OptimizationAlgorithm alg, bool use_gradient_accumulation, in GetOptimizationAlgorithmStateVariables()
122 def __init__(self, learning_rate, use_gradient_accumulation): argument131 use_gradient_accumulation=True): argument158 use_gradient_accumulation=True): argument