CondVAEModel.on_save_checkpoint

CondVAEModel.on_save_checkpoint(checkpoint: Dict[str, Any]) → None

Save the extra hyperparameters to the model checkpoint.

Parameters:

checkpoint (Dict[str, Any]) – The full checkpoint dictionary before it gets dumped to a file.