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
487 stars 17 forks source link

Johnmark savegradients #5

Closed johnmarktaylor91 closed 1 year ago

johnmarktaylor91 commented 1 year ago

Merging the functionality to save gradients during the backward pass.