VEncoder.summary

VEncoder.summary(detail: int = 1, flag_print: bool = True) None | str[source]

Prints a summary of a PyTorch model, including the number of parameters, the layers, their names, and the dimensionality.

Parameters:
  • detail (int, optional, default=1) – Controls the level of detail in the summary. Set to 1 to print the name and dimensionality of each layer.

  • flag_print (bool, optional, default=True) – If flag_print is True, prints the summary. Otherwise, returns a string containing the summary.

Returns:

Union[None, str] – If flag_print is True, prints the summary. Otherwise, returns a string containing the summary.