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