Skip to content

Commit 0d175bf

Browse files
authored
Update README.md
1 parent 521fd65 commit 0d175bf

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

README.md

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
# pytorch_training_optimization_using_tensordict_memory_mapping
2-
Optimizing PyTorch Training using TensorDict Memory Mapping to Nvidia GPU.
2+
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.
35

46
````
57
python==3.9.13

0 commit comments

Comments
 (0)