Skip to content

Commit

Permalink
Merge pull request #99 from chengduoZH/update_se_resnext
Browse files Browse the repository at this point in the history
Update SE-Resnext
  • Loading branch information
chengduo committed May 18, 2018
2 parents 619ff37 + 640677d commit 9eec3bf
Show file tree
Hide file tree
Showing 2 changed files with 137 additions and 134 deletions.
18 changes: 9 additions & 9 deletions fluid/SE-ResNeXt-152/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,28 +2,28 @@

## For single card:
```
env CUDA_VISIBLE_DEVICES=0 python train.py --use_parallel_mode=parallel_do --do_profile=False --use_nccl=False --parallel=False --display_step=1
env CUDA_VISIBLE_DEVICES=0 python train.py --parallel_mode=parallel_do --display_step=1
```

## For multi-card:
### use parallel_do (the data is in GPU side.)
```
env CUDA_VISIBLE_DEVICES=4,5,6,7 python train.py --use_parallel_mode=parallel_do --do_profile=False --use_nccl=True --parallel=True --use_python_reader=false --display_step=1
env CUDA_VISIBLE_DEVICES=4,5,6,7 python train.py --parallel_mode=parallel_do --fix_data_in_gpu=true --display_step=1
```
### use parallel_do (use_python_reader, data_flow: CPU->GPU->Training)
### use parallel_do (data_flow: CPU->GPU->Training)
```
env CUDA_VISIBLE_DEVICES=4,5,6,7 python train.py --use_parallel_mode=parallel_do --do_profile=False --use_nccl=True --parallel=True --use_python_reader=true --display_step=1
env CUDA_VISIBLE_DEVICES=4,5,6,7 python train.py --parallel_mode=parallel_do --display_step=1
```

### use parallel_exe (use C++ reader)
```
env CUDA_VISIBLE_DEVICES=4,5,6,7 python train.py --use_parallel_mode=parallel_exe --do_profile=False --display_step=1
env CUDA_VISIBLE_DEVICES=4,5,6,7 python train.py --parallel_mode=parallel_exe --use_recordio=True --display_step=1
```
#### use parallel_exe (use feeder, the data is in GPU side)
#### use parallel_exe (the data is in GPU side)
```
env CUDA_VISIBLE_DEVICES=4,5,6,7 python train.py --use_parallel_mode=parallel_exe --use_feeder=true --use_python_reader=false --do_profile=false --display_step=1
env CUDA_VISIBLE_DEVICES=4,5,6,7 python train.py --parallel_mode=parallel_exe --fix_data_in_gpu=true --display_step=1
```
#### use parallel_exe (use feeder and use_python_readear, data flow: CPU->GPU->Training)
#### use parallel_exe (data flow: CPU->GPU->Training)
```
env CUDA_VISIBLE_DEVICES=4,5,6,7 python train.py --use_parallel_mode=parallel_exe --use_feeder=true --use_python_reader=true --do_profile=false --display_step=1
env CUDA_VISIBLE_DEVICES=4,5,6,7 python train.py --parallel_mode=parallel_exe --display_step=1
```
Loading

0 comments on commit 9eec3bf

Please sign in to comment.