coral-pytorch icon indicating copy to clipboard operation
coral-pytorch copied to clipboard

save the second logsigmoid computation

Open nlassaux opened this issue 1 year ago • 0 comments

The computation of the logsigmoid can be done once and stored to save some CPU/GPU cycles. In both coral_loss and corn_loss

nlassaux avatar Jul 06 '23 17:07 nlassaux