Searched refs:skip_target_indices (Results 1 – 1 of 1) sorted by relevance
321 skip_target_indices = []325 skip_target_indices.append(i)350 self._set_metric_attributes(skip_target_indices=skip_target_indices)388 if i in skip_target_indices:424 self._set_metric_attributes(skip_target_indices=skip_target_indices)431 skip_target_indices=skip_target_indices,439 self.total_loss = self._prepare_total_loss(skip_target_indices, masks)1630 def _prepare_total_loss(self, skip_target_indices=None, masks=None): argument1647 skip_target_indices = skip_target_indices or []1654 if i in skip_target_indices:[all …]