Searched refs:ShampooOptimizer (Results 1 – 3 of 3) sorted by relevance
64 opt = shampoo.ShampooOptimizer(global_step)115 opt = shampoo.ShampooOptimizer(global_step)174 opt = shampoo.ShampooOptimizer(global_step,264 opt = shampoo.ShampooOptimizer(global_step)320 opt = shampoo.ShampooOptimizer(global_step)399 opt = shampoo.ShampooOptimizer(global_step)475 opt = shampoo.ShampooOptimizer(global_step,543 opt = shampoo.ShampooOptimizer(global_step, gbar_decay=gbar_decay,637 opt = shampoo.ShampooOptimizer(global_step, svd_interval=svd_interval,714 opt = shampoo.ShampooOptimizer(
45 class ShampooOptimizer(optimizer.Optimizer): class122 super(ShampooOptimizer, self).__init__(use_locking, name)
386 shampoo.ShampooOptimizer):