nnsight
nnsight copied to clipboard
Nested Value Injection
Handle value injection recursively for Collection data defined outside of tracing.
e.g.
arr = list()
with tiny_model.trace(tiny_input):
arr.append(tiny_model.output.save())
assert isinstance(arr[0], torch.Tensor)