Searched defs:inner_optimizer (Results 1 – 2 of 2) sorted by relevance
93 def __init__(self, inner_optimizer, loss_scale): argument111 def __init__(self, inner_optimizer, loss_scale): argument117 def create_lso(inner_optimizer, loss_scale): argument
402 def __init__(self, inner_optimizer, dynamic=True, initial_scale=None, argument510 def inner_optimizer(self): member in LossScaleOptimizer