"vscode:/vscode.git/clone" did not exist on "2316ca1f748a9e104f0dfefe99b6b8d69f1af445"
Commit 3bc7f6c2 authored by Alexandr Baranezky's avatar Alexandr Baranezky
Browse files

1. Little bit optimized code

2. Fixed accuracy when calcuting logs more than 10 steps
parent 3be9ece9
...@@ -53,6 +53,9 @@ tf.app.flags.DEFINE_integer('max_steps', 1000000, ...@@ -53,6 +53,9 @@ tf.app.flags.DEFINE_integer('max_steps', 1000000,
tf.app.flags.DEFINE_boolean('log_device_placement', False, tf.app.flags.DEFINE_boolean('log_device_placement', False,
"""Whether to log device placement.""") """Whether to log device placement.""")
tf.app.flags.DEFINE_integer('log_steps_count', 10,
"""Log process results per count.""")
def train(): def train():
"""Train CIFAR-10 for a number of steps.""" """Train CIFAR-10 for a number of steps."""
...@@ -78,19 +81,23 @@ def train(): ...@@ -78,19 +81,23 @@ def train():
def begin(self): def begin(self):
self._step = -1 self._step = -1
self._start_time = time.time()
def before_run(self, run_context): def before_run(self, run_context):
self._step += 1 self._step += 1
self._start_time = time.time()
return tf.train.SessionRunArgs(loss) # Asks for loss value. return tf.train.SessionRunArgs(loss) # Asks for loss value.
def after_run(self, run_context, run_values): def after_run(self, run_context, run_values):
log_steps = FLAGS.log_steps_count
if self._step % log_steps == 0:
duration = time.time() - self._start_time duration = time.time() - self._start_time
self._start_time = time.time()
loss_value = run_values.results loss_value = run_values.results
if self._step % 10 == 0:
num_examples_per_step = FLAGS.batch_size num_examples_per_step = FLAGS.batch_size
examples_per_sec = num_examples_per_step / duration examples_per_sec = num_examples_per_step * FLAGS.log_steps_count / duration
sec_per_batch = float(duration) sec_per_batch = float(duration / log_steps)
format_str = ('%s: step %d, loss = %.2f (%.1f examples/sec; %.3f ' format_str = ('%s: step %d, loss = %.2f (%.1f examples/sec; %.3f '
'sec/batch)') 'sec/batch)')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment