Home
last modified time | relevance | path

Searched refs:current_step (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_distributed.py386 current_step = 0
387 while current_step < target_steps:
388 step_count = steps_to_run[current_step] if use_steps else 1
404 target_steps = current_step
414 current_step += 1
531 current_step = 0
532 while current_step < target_steps:
533 batch_logs = {'batch': current_step, 'size': 1}
534 callbacks._call_batch_hook(mode, 'begin', current_step, batch_logs)
542 warning_msg = 'Number of steps ran: {} steps'.format(current_step)
[all …]
/external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/
Dmain_estimator_tpu.py248 current_step = int(os.path.basename(ckpt).split("-")[1])
249 if current_step >= revnet_config.max_train_iter:
251 "Evaluation finished after training step %d" % current_step)
263 current_step = estimator._load_global_step_from_checkpoint_dir(
270 current_step))
281 while current_step < revnet_config.max_train_iter:
284 next_checkpoint = min(current_step + FLAGS.steps_per_eval,
288 current_step = next_checkpoint
/external/tensorflow/tensorflow/lite/experimental/micro/examples/micro_speech/
Dfeature_provider.cc46 const int current_step = (time_in_ms / kFeatureSliceStrideMs); in PopulateFeatureData() local
48 int slices_needed = current_step - last_step; in PopulateFeatureData()
94 const int new_step = (current_step - kFeatureSliceCount + 1) + new_slice; in PopulateFeatureData()
/external/swiftshader/third_party/SPIRV-Tools/source/reduce/
Dreducer.cpp29 bool ReachedStepLimit(uint32_t current_step,
148 bool Reducer::Impl::ReachedStepLimit(uint32_t current_step, in ReachedStepLimit() argument
150 return current_step >= options->step_limit; in ReachedStepLimit()
/external/deqp-deps/SPIRV-Tools/source/reduce/
Dreducer.cpp29 bool ReachedStepLimit(uint32_t current_step,
148 bool Reducer::Impl::ReachedStepLimit(uint32_t current_step, in ReachedStepLimit() argument
150 return current_step >= options->step_limit; in ReachedStepLimit()
/external/tensorflow/tensorflow/python/training/
Dsupervisor.py676 current_step = training_util.global_step(sess, self._global_step)
679 current_step)
1071 current_step = training_util.global_step(self._sess, self._step_counter)
1072 added_steps = current_step - self._last_step
1073 self._last_step = current_step
1086 self._sv.summary_writer.add_summary(summary, current_step)
1110 current_step = training_util.global_step(self._sess, self._sv.global_step)
1114 current_step)
Dbasic_session_run_hooks.py906 current_step = run_context.session.run(self._global_step_tensor)
907 if current_step >= self._wait_until_step:
910 if current_step - last_logged_step > 1000:
912 "Current step is %d.", self._wait_until_step, current_step)
913 last_logged_step = current_step
/external/tensorflow/tensorflow/contrib/tensor_forest/client/
Drandom_forest.py95 current_step = run_values.results['global_step']
99 if self.last_step == -1 or self.last_step > current_step:
101 self.last_step = current_step
106 self.last_step = current_step
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Dtrainer_hooks.py210 def _get_train_op_for_global_step(self, current_step): argument
212 if current_step < self._train_steps:
/external/tensorflow/tensorflow/contrib/learn/python/learn/
Dmonitors.py1199 def every_n_step_end(self, current_step, outputs): argument
1202 added_steps = current_step - self._last_reported_step
1208 self._summary_writer.add_summary(summary, current_step)
1209 self._last_reported_step = current_step