def run(self, fetches, feed_dict=None, options=None, run_metadata=None): """See base class.""" if self.should_stop(): raise RuntimeError('Run called even after should_stop requested.') actual_fetches = {'caller': fetches} run_context = session_run_hook.SessionRunContext( original_args=session_run_hook.SessionRunArgs(fetches, feed_dict), session=self._sess) options = options or config_pb2.RunOptions() feed_dict = self._call_hook_before_run(run_context, actual_fetches, feed_dict, options) # Do session run. run_metadata = run_metadata or config_pb2.RunMetadata() outputs = _WrappedSession.run(self, fetches=actual_fetches, feed_dict=feed_dict, options=options, run_metadata=run_metadata) for hook in self._hooks: hook.after_run( run_context, session_run_hook.SessionRunValues( results=outputs[hook] if hook in outputs else None, options=options, run_metadata=run_metadata)) self._should_stop = self._should_stop or run_context.stop_requested return outputs['caller']
def run_hook_with_indices(self, sweep_hook, row_indices, col_indices): with self.test_session() as sess: # Before run. run_context = session_run_hook.SessionRunContext( original_args=None, session=sess) sess_run_args = sweep_hook.before_run(run_context) feed_dict = { self._input_row_indices_ph: row_indices, self._input_col_indices_ph: col_indices } # Run. run_results = sess.run(sess_run_args.fetches, feed_dict=feed_dict) run_values = session_run_hook.SessionRunValues( results=run_results, options=None, run_metadata=None) # After run. sweep_hook.after_run(run_context, run_values)