Searched defs:initial_scale (Results 1 – 1 of 1) sorted by relevance
514 def __init__(self, inner_optimizer, dynamic=True, initial_scale=None, argument596 def initial_scale(self): member in LossScaleOptimizer