Searched refs:_weighted_loss_merger (Results 1 – 1 of 1) sorted by relevance
485 def _weighted_loss_merger(losses): function496 return _MultiHead(heads, loss_merger=_weighted_loss_merger)