SpeechT5/SpeechLM/speechlm/criterions/__init__.py

import importlib
import os

for file in os.listdir(os.path.dirname(__file__)):
    if file.endswith(".py") and not file.startswith("_"):
        criterion_name = file[: file.find(".py")]
        importlib.import_module(
            "speechlm.criterions." + criterion_name
        )