Skip to content

Commit eb6e50e

Browse files
author
Soheil Bahrampour
committed
fixed a bug in timing as suggested here: https://goo.gl/Y4a4Gd
1 parent e96c4df commit eb6e50e

File tree

3 files changed

+9
-4
lines changed

3 files changed

+9
-4
lines changed

tensorflow/LeNet/lenet.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,9 +106,10 @@ def time_tensorflow_run(session, target, info_string, config):
106106
processing_times = np.zeros(config.num_timing_iters)
107107
if not isinstance(target, list):
108108
target = [target]
109+
target_op = tf.group(*target)
109110
for i in xrange(config.num_timing_iters + config.num_warmup_iters):
110111
start_time = time.time()
111-
_ = session.run(tf.group(*target))
112+
_ = session.run(target_op)
112113
duration = time.time() - start_time
113114
if i >= config.num_warmup_iters:
114115
processing_times[i-config.num_warmup_iters] = duration

tensorflow/alexnet/alexnet.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -121,9 +121,10 @@ def time_tensorflow_run(session, target, info_string, config):
121121
processing_times = np.zeros(config.num_timing_iters)
122122
if not isinstance(target, list):
123123
target = [target]
124+
group_op = tf.group(*target)
124125
for i in xrange(config.num_timing_iters + config.num_warmup_iters):
125126
start_time = time.time()
126-
_ = session.run(tf.group(*target))
127+
_ = session.run(group_op)
127128
duration = time.time() - start_time
128129
if i >= config.num_warmup_iters:
129130
processing_times[i-config.num_warmup_iters] = duration
@@ -164,7 +165,9 @@ def time_alexnet():
164165
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=
165166
config.gpu_memory_fraction)
166167
device_config = tf.ConfigProto(gpu_options=gpu_options,
167-
device_count={'GPU': 1})
168+
device_count={'GPU': 1},
169+
#log_device_placement=True
170+
)
168171
print device_config
169172
sess = tf.Session(config=device_config)
170173
sess.run(init)

tensorflow/stackedAE/stackedAE.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -142,9 +142,10 @@ def time_tensorflow_run(session, target, info_string, config):
142142
processing_times = np.zeros(config.num_timing_iters)
143143
if not isinstance(target, list):
144144
target = [target]
145+
group_op = tf.group(*target)
145146
for i in xrange(config.num_timing_iters + config.num_warmup_iters):
146147
start_time = time.time()
147-
_ = session.run(tf.group(*target))
148+
_ = session.run(group_op)
148149
duration = time.time() - start_time
149150
if i >= config.num_warmup_iters:
150151
processing_times[i-config.num_warmup_iters] = duration

0 commit comments

Comments
 (0)