We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 521fd65 commit 0d175bfCopy full SHA for 0d175bf
README.md
@@ -1,5 +1,7 @@
1
# pytorch_training_optimization_using_tensordict_memory_mapping
2
-Optimizing PyTorch Training using TensorDict Memory Mapping to Nvidia GPU.
+
3
+Optimizing PyTorch training by wrapping torch.utils.data.Dataset with tensordict.TensorDict.MemoryMappedTensor mapped, pinned,
4
+and loaded onto an Nvidia GPU and inputting TensorDict(Dataset) into torch.utils.data.DataLoader--to boost model training speed.
5
6
````
7
python==3.9.13
0 commit comments