search

johnmarktaylor91 / torchlens

Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.
GNU General Public License v3.0
456 stars 16 forks source link

Repeated Buffer Passes #17

Closed johnmarktaylor91 closed 11 months ago

johnmarktaylor91 commented 11 months ago

Fixed an edge case of models that change buffer tensors during the forward pass.