Performance, Pytorch, Deep Learning, Artificial Intelligence, Tensorboard.
"background" labels to the ignore index with torch.profiler.record_function('index assignment'): target[inds] = self.ignore_val return target def forward(self, pred: Tensor, target: Tensor) -> Tensor: # ignore background labels target = self.ignore_background(target) # retrieve a list of unique