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