Spaces:
Running
on
T4
Running
on
T4
// Interface for Python | |
at::Tensor SigmoidFocalLoss_forward( | |
const at::Tensor& logits, | |
const at::Tensor& targets, | |
const int num_classes, | |
const float gamma, | |
const float alpha) { | |
if (logits.device().is_cuda()) { | |
return SigmoidFocalLoss_forward_cuda(logits, targets, num_classes, gamma, alpha); | |
AT_ERROR("Not compiled with GPU support"); | |
} | |
AT_ERROR("Not implemented on the CPU"); | |
} | |
at::Tensor SigmoidFocalLoss_backward( | |
const at::Tensor& logits, | |
const at::Tensor& targets, | |
const at::Tensor& d_losses, | |
const int num_classes, | |
const float gamma, | |
const float alpha) { | |
if (logits.device().is_cuda()) { | |
return SigmoidFocalLoss_backward_cuda(logits, targets, d_losses, num_classes, gamma, alpha); | |
AT_ERROR("Not compiled with GPU support"); | |
} | |
AT_ERROR("Not implemented on the CPU"); | |
} | |