PyTorch apply

A function that applies a given function to all submodules of an nn.Module, similar to map.

It traverses modules using postorder traversal, prioritizing the left child.

1
def do_something(m):
2
  # do something!
3
  return m
4
model = #something very complex model
5
result = model.apply(print_module)