Menu

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_smcvae(checkpointfile, n_channels, n_feats, **kwargs)[source]ΒΆ

Return the sMCVAE model.

Parameters

checkpointfile : str

path to the model weights.

latent_dim : int

the number of latent dimensions.

n_channels : int

the number of input channels/views.

n_feats : list of int

the number of features for each channel.

kwargs : dict

extra parameters passed to the MCVAE constructor.

Returns

model : Module

instanciated model.

Follow us

© 2023, mmbench developers