CondVAEModel.summary
- CondVAEModel.summary(max_depth: int = 1, flag_print: bool = True) str | None [source]
Prints a summary of the encoder and decoder, including the number of parameters, the layers, their names, and the dimensionality.
- Parameters:
max_depth (int, optional, default=1) – Maximum depth of modules to show. Use -1 to show all modules or 0 to show no summary.
flag_print (bool, optional, default=True) – If True, print the summary to the console. Otherwise, return the summary as a string.