Spaces:
Sleeping
Sleeping
# Copyright (c) OpenMMLab. All rights reserved. | |
import torch.nn as nn | |
from mmocr.registry import MODELS | |
class CrossEntropyLoss(nn.CrossEntropyLoss): | |
"""Cross entropy loss.""" | |