File size: 690 Bytes
c668e80 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
"""Module defining various utilities."""
from onmt.utils.misc import use_gpu, set_random_seed
from onmt.utils.alignment import make_batch_align_matrix
from onmt.utils.report_manager import ReportMgr, build_report_manager
from onmt.utils.statistics import Statistics
from onmt.utils.optimizers import MultipleOptimizer, Optimizer, AdaFactor, FusedAdam
from onmt.utils.earlystopping import EarlyStopping, scorers_from_opts
__all__ = [
"use_gpu",
"set_random_seed",
"ReportMgr",
"build_report_manager",
"Statistics",
"MultipleOptimizer",
"Optimizer",
"AdaFactor",
"FusedAdam",
"EarlyStopping",
"scorers_from_opts",
"make_batch_align_matrix",
]
|