D-FINE/src/nn/criterion/__init__.py

"""
Copied from RT-DETR (https://github.com/lyuwenyu/RT-DETR)
Copyright(c) 2023 lyuwenyu. All Rights Reserved.
"""

import torch.nn as nn

from ...core import register
from .det_criterion import DetCriterion

CrossEntropyLoss = register()(nn.CrossEntropyLoss)