Searched defs:loss_scale_manager (Results 1 – 3 of 3) sorted by relevance
22 def train_wrap(net, loss_fn=None, optimizer=None, weights=None, loss_scale_manager=None): argument
36 loss_scale_manager = FixedLossScaleManager(loss_scale, False) variable
182 loss_scale_manager=None, keep_batchnorm_fp32=False): argument