From 3754beee8ae3d525cb358cf964131309b76fd05a Mon Sep 17 00:00:00 2001 From: Kaiyu Shi Date: Fri, 26 Jul 2024 09:34:31 +0000 Subject: [PATCH] Change to UntypedStorage in README --- README.md | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 76ce638..ab34eb3 100644 --- a/README.md +++ b/README.md @@ -204,9 +204,12 @@ As *Memory Profiler* only gives the overall memory usage information by lines, a more low-level memory usage information can be obtained by *Memory Reporter*. *Memory reporter* iterates all the `Tensor` objects and gets the underlying -`Storage` object to get the actual memory usage instead of the surface +`UntypedStorage` (previously `Storage`) object to get the actual memory usage instead of the surface `Tensor.size`. +> see [UntypedStorage](https://pytorch.org/docs/stable/storage.html#torch.UntypedStorage) for detailed +> information + #### Sample - A minimal one: