Searched refs:loss_merger (Results 1 – 1 of 1) sorted by relevance
496 return _MultiHead(heads, loss_merger=_weighted_loss_merger)1569 def __init__(self, heads, loss_merger): argument1587 self._loss_merger = loss_merger