disvae/__init__.py

from disvae.evaluate import Evaluator
from disvae.main import get_kl_dict
from disvae.training import Trainer
from disvae.utils.modelIO import load_model, save_model

# from disvae.models.vae import init_specific_model  # notwendig?