Python class

PrintHook

`PrintHook`

class max.nn.hooks.PrintHook(export_path=None, filter=None)

source

Bases: BasePrintHook

Hook that prints/saves layer tensor inputs and outputs.

This class must be initialized added before the graph is built so the print ops can be added to the graph.

Parameters:

property export_path: str | None

source

name_layers(model)

source

Create names for all layers in the model based on nested attributes.

name_layers_v3(model)

source

Create names for all v3 Module layers based on nested attributes.

print_value(name, value)

source

Prints a value, and returns whether the print is successful.

Parameters:

Return type:

bool

remove()

source