Searched defs:initial_loss_scale (Results 1 – 3 of 3) sorted by relevance
350 initial_loss_scale=2 ** 15, # See docstring for why this is big. argument382 def initial_loss_scale(self): member in DynamicLossScale
127 initial_loss_scale=1., argument
216 initial_loss_scale, argument300 def initial_loss_scale(self): member in _DynamicLossScaleState