Performance, Pytorch, Deep Learning, Artificial Intelligence, Tensorboard.
torch.profiler.record_function('create mask'): mask = target != self.ignore_val # permute the logits in preparation for masking with torch.profiler.record_function('permute'): permuted_pred = torch.permute(pred, [0, 2, 3, 1]) # apply the boolean mask to the targets and logits with