Skip to content

Commit 9578694

Browse files
authored
Data Loader Test README
1 parent 8f6cf11 commit 9578694

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

README.md

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ CUDA_VISIBLE_DEVICES='0,1' python -m src.tools.check_dist --num-gpu 2 --num-mach
3737
```
3838

3939
## Test
40-
Examples for collective communication functions in single machine.
40+
Examples for collective communication functions and data loader in single machine.
4141
It also can be executed in multi-machine settings.
4242
### Gather
4343
```bash
@@ -48,3 +48,8 @@ CUDA_VISIBLE_DEVICES='0,1' python -m src.tools.gather --num-gpu 2
4848
```bash
4949
CUDA_VISIBLE_DEVICES='0,1' python -m src.tools.reduce --num-gpu 2
5050
```
51+
52+
### Data Loader
53+
```bash
54+
CUDA_VISIBLE_DEVICES='0,1' python -m src.tools.loader --num-gpu 2 --seed 0
55+
```

0 commit comments

Comments
 (0)