mmocr/models/common/losses/ce_loss.py

# 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."""