Skip to content

Commit 3b28e8f

Browse files
authored
Update README.md
1 parent 5234e27 commit 3b28e8f

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

README.md

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,9 @@
11
# pytorch_training_optimization_using_tensordict_memory_mapping
2-
Optimizing PyTorch Training using TensorDict Memory Mapping
2+
Optimizing PyTorch Training using TensorDict Memory Mapping.
3+
4+
To run the demo: python run_demo.py
5+
6+
[NOTE: a mock dataset will be created in the current working directory.]
37

48

59
Training 1 Epoch via torch.utils.data.Dataset:

0 commit comments

Comments
 (0)