codecommentclassification/modeling/transformer/__init__.py

"""Transformer model trainer module."""

import logging

from .trainer import TransformerConfig, TransformerTrainer

logger = logging.getLogger(__name__)
logger.addHandler(logging.NullHandler())

__all__ = ["TransformerConfig", "TransformerTrainer"]