Performance, Pytorch, Deep Learning, Artificial Intelligence, Tensorboard.
torch.where(target==self.num_classes-1, -1*torch.ones_like(target),target) return target In the image below we show the Trace View following this change. Trace View Following Optimization #1 (Captured by Author) While we have succeeded in removing the cudaMempyAsync coming from the torch.nonzero