mogen/models/losses/__init__.py

from .mse_loss import MSELoss, KinematicLoss
from .utils import (convert_to_one_hot, reduce_loss, weight_reduce_loss,
                    weighted_loss)

__all__ = [
    'convert_to_one_hot', 'reduce_loss', 'weight_reduce_loss', 'weighted_loss',
    'MSELoss', 'KinematicLoss'
]