Spaces:
Running
on
Zero
Running
on
Zero
File size: 271 Bytes
e85fecb |
1 2 3 4 5 6 7 8 9 10 11 12 |
"""
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)
|