training/tacotron2_model/__init__.py

from training.tacotron2_model.model import Tacotron2  # noqa
from training.tacotron2_model.loss import Tacotron2Loss  # noqa
from training.tacotron2_model.collate import TextMelCollate  # noqa
from training.tacotron2_model.stft import TacotronSTFT  # noqa