# from .fhvae import FHVAE
# from .simple_fhvae import SimpleFHVAE
# from .reg_fhvae import RegFHVAE
#
# def load_model(name):
#     if name == "fhvae":
#         return FHVAE
#     elif name == "simple_fhvae":
#         return SimpleFHVAE
#     elif name == "reg_fhvae":
#         return RegFHVAE
#     else:
#         raise ValueError