Spaces:
Sleeping
Sleeping
File size: 208 Bytes
9bf4bd7 |
1 2 3 4 5 6 7 8 9 10 |
# Copyright (c) OpenMMLab. All rights reserved.
import torch.nn as nn
from mmocr.registry import MODELS
@MODELS.register_module()
class CrossEntropyLoss(nn.CrossEntropyLoss):
"""Cross entropy loss."""
|