Neural Network Abstractions: Own Parameters, Preserve the Graph
Two structures, two lifetimes
The module structure owns reusable parameter leaves.
The forward pass creates a new temporary computation graph.
Predict what fails if weights are created inside forward.
1 / 8