load_pretrained_ae¶

behavenet.models.aes.load_pretrained_ae(model, hparams)[source]¶

Load pretrained weights into already constructed AE model.

Parameters:

model (behavenet.models.aes object) – autoencoder-based model; AE, ConditionalAE, AEMSP
hparams (dict) – needs to contain keys model_type and pretrained_weights_path

Returns:

input model with updated weights

Return type:

behavenet.models.aes object