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.