File size: 350 Bytes
c1a46f7
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
"""训练模块 (Person C 负责)"""

from easytranslate.training.trainer import Trainer
from easytranslate.training.optimizer import build_optimizer, build_scheduler
from easytranslate.training.loss import LabelSmoothedCrossEntropyLoss

__all__ = [
    "Trainer",
    "build_optimizer",
    "build_scheduler",
    "LabelSmoothedCrossEntropyLoss",
]