Benchmark multi-model/multi-view models.
Note
This page is a reference documentation. It only explains the function signature, and not how to use it. Please refer to the gallery for the big picture.
-
mmbench.model.get_vae(checkpointfile, **kwargs)[source]ΒΆ Return the VAE model.
- Parameters
checkpointfiles : str
path to the model weights.
latent_dim : int
the number of latent dimensions.
input_dim : int
the number of features for the image channel.
kwargs : dict
extra parameters passed to the NeuroCLAV constructor.
- Returns
model : Module
the instanciated model.
Follow us