coral-pytorch
coral-pytorch copied to clipboard
save the second logsigmoid computation
The computation of the logsigmoid can be done once and stored to save some CPU/GPU cycles. In both coral_loss
and corn_loss