Skip to main content

util.model_util

logger_config

logger

load_model

def load_model(cfg: dict, model_name: str, fixed: bool = True) -> RNN_VAE

Load the VAME model.

Args: cfg (dict): Configuration dictionary. model_name (str): Name of the model. fixed (bool): Fixed or variable length sequences.

Returns RNN_VAE: Loaded VAME model.