nnsight icon indicating copy to clipboard operation
nnsight copied to clipboard

Nested Value Injection

Open AdamBelfki3 opened this issue 8 months ago • 0 comments

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)

AdamBelfki3 avatar Apr 04 '25 02:04 AdamBelfki3