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