Optimization, Pytorch, Machine Learning, Deep Learning, Tensorboard.
and rerun the profiling analysis. # custom loss definition class CrossEntropyLoss(nn.Module): def forward(self, input, target): with torch.profiler.record_function('log_softmax'): pred = log_softmax(input) with torch.profiler.record_function('define_weights'): weights =