Searched refs:gbar_decay (Results 1 – 3 of 3) sorted by relevance
396 gbar_decay=0.0, argument442 gbar_decay=gbar_decay,
74 gbar_decay=0.0, argument126 self._gbar_decay = gbar_decay
532 gbar_decay = 0.9543 opt = shampoo.ShampooOptimizer(global_step, gbar_decay=gbar_decay,592 gbar_np_2 = gbar_decay * gbar_np + gbar_weight * grad_np_2