Calls tensor.untyped_storage instead of tesnor.storage - Githubissues

Stonesjtu / pytorch_memlab

Profiling and inspecting memory in pytorch

MIT License

1.01k stars 37 forks source link

Calls tensor.untyped_storage instead of tesnor.storage #59

Closed wangkuiyi closed 1 month ago

wangkuiyi commented 2 months ago

Fix https://github.com/Stonesjtu/pytorch_memlab/issues/58

Stonesjtu commented 1 month ago

LGTM overall. Can you update the README for using UntypedStorage class here: https://github.com/Stonesjtu/pytorch_memlab/blob/212e59568e72e4ea6275b38031f4f766bf3d70d0/README.md?plain=1#L207