CondAEModel.on_load_checkpoint

CondAEModel.on_load_checkpoint(checkpoint: Dict[str, Any]) None[source]

Load the extra hyperparameters from the model checkpoint. The data module parameters are also loaded, if they exist.

Parameters:

checkpoint (Dict[str, Any]) – Loaded checkpoint.