Skip to content

Commit 11276f2

Browse files
authored
Merge pull request #41 from dzhwinter/feature/rnn
refine file structure
2 parents 5b3ab61 + 898a545 commit 11276f2

File tree

7 files changed

+2
-465
lines changed

7 files changed

+2
-465
lines changed

HowToDoBenchmark.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ docker pull tensorflow/tensorflow:1.4.0-gpu
3838

3939
首先需要屏蔽GPU `export CUDA_VISIBLE_DEVICES=`;
4040

41-
Fluid需要关闭OpenMP. 设置`export OMP_NUM_THREADS=1`. 设置Device=CPU, 或者在代码中设置CPUPlace().
41+
在单机单卡的测试环境中,Fluid需要关闭OpenMP. 设置`export OMP_NUM_THREADS=1`. 设置Device=CPU, 或者在代码中设置CPUPlace().
4242
TensorFlow需要关闭多线程, 设置 intra_op_parallelism_threads=1, inter_op_parallelism_threads=1.
4343
运行过程中可以通过, `nvidia-smi`来校验是否有GPU被使用, 下文GPU同理.
4444

File renamed without changes.
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ def gate_common(
113113
adam = fluid.optimizer.Adam()
114114
adam.minimize(loss)
115115

116-
place = fluid.CPUPlace() if args.device == 'CPU' else fluid.GPUPlace(0)
116+
place = fluid.CPUPlace() if args.device == 'CPU' else fluid.CUDAPlace(0)
117117
exe = fluid.Executor(place)
118118
exe.run(fluid.default_startup_program())
119119

fluid/understand_sentiment_dynamic_lstm.py

Lines changed: 0 additions & 157 deletions
This file was deleted.

fluid/understand_sentiment_lstm.py

Lines changed: 0 additions & 179 deletions
This file was deleted.

0 commit comments

Comments
 (0)