Encoder.summary
- Encoder.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.