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