Searched refs:estimator_spec (Results 1 – 13 of 13) sorted by relevance
65 self, model_fn_ops, estimator_spec): argument70 self.assertEqual(model_fn_ops.predictions, estimator_spec.predictions)71 self.assertEqual(model_fn_ops.loss, estimator_spec.loss)72 self.assertEqual(model_fn_ops.train_op, estimator_spec.train_op)74 estimator_spec.eval_metric_ops)76 estimator_spec.training_chief_hooks)78 estimator_spec.training_hooks)79 self.assertEqual(model_fn_ops.scaffold, estimator_spec.scaffold)86 estimator_spec = model_fn_ops.estimator_spec()87 self.assertEquals_except_export_and_eval_loss(model_fn_ops, estimator_spec)[all …]
212 def estimator_spec(self, default_serving_output_alternative_key=None): member in ModelFnOps
61 def estimator_spec_to_model_fn_ops(estimator_spec, export_alternatives=False): argument64 estimator_spec.export_outputs)69 mode=_core_mode_to_contrib_mode(estimator_spec.mode),70 predictions=estimator_spec.predictions,71 loss=estimator_spec.loss,72 train_op=estimator_spec.train_op,73 eval_metric_ops=estimator_spec.eval_metric_ops,
182 estimator_spec = head.create_estimator_spec(189 estimator_spec = estimator_spec._replace(190 training_hooks=training_hooks + list(estimator_spec.training_hooks))191 return estimator_spec402 estimator_spec = head.create_estimator_spec(409 estimator_spec = estimator_spec._replace(410 training_hooks=training_hooks + list(estimator_spec.training_hooks))411 return estimator_spec
315 estimator_spec = model_fn(319 self.assertEqual(estimator_spec.mode, _TRAIN)322 self.assertEqual(sess.run(estimator_spec.loss), sess.run(loss))323 self.assertEqual(sess.run(estimator_spec.train_op), sess.run(loss))339 estimator_spec = model_fn(343 self.assertEqual(estimator_spec.mode, _TRAIN)346 self.assertEqual(sess.run(estimator_spec.loss), sess.run(loss))347 self.assertEqual(sess.run(estimator_spec.train_op), sess.run(loss))363 estimator_spec = model_fn_with_summary(367 self.assertEqual(sess.run(estimator_spec.loss), _EXPECTED_LOSS)[all …]
618 estimator_spec = self._call_model_fn(features, labels,622 captured_scaffold_fn.capture(estimator_spec.scaffold_fn)631 with ops.control_dependencies([estimator_spec.train_op]):632 return array_ops.identity(estimator_spec.loss)643 estimator_spec = self._call_model_fn(features, labels,647 captured_scaffold_fn.capture(estimator_spec.scaffold_fn)654 if estimator_spec.eval_metrics:655 (eval_metric_fn, eval_metric_fn_tensors) = estimator_spec.eval_metrics669 return tuple([estimator_spec.loss] + eval_metric_fn_tensors)692 def _verify_estimator_spec(self, estimator_spec): argument[all …]
295 estimator_spec = model_head.create_estimator_spec(307 loss_op=estimator_spec.loss))309 estimator_spec = estimator_spec._replace(310 training_hooks=training_hooks + list(estimator_spec.training_hooks))312 estimator_spec.predictions[keys_name] = keys314 estimator_spec.predictions[TREE_PATHS_PREDICTION_KEY] = tree_paths315 estimator_spec.predictions[VARIANCE_PREDICTION_KEY] = regression_variance318 outputs = estimator_spec.export_outputs321 outputs = {ALL_SERVING_KEY: PredictOutput(estimator_spec.predictions)}322 print(estimator_spec.export_outputs)[all …]
242 estimator_spec = _get_estimator_spec(246 assert isinstance(estimator_spec, tpu_estimator.TPUEstimatorSpec)247 return estimator_spec279 estimator_spec = tpu_estimator.TPUEstimatorSpec(293 estimator_spec = _get_eval_estimator_spec(309 estimator_spec = _get_train_estimator_spec(312 return estimator_spec
224 estimator_spec = model_fn_lib.EstimatorSpec(233 estimator_spec = _get_eval_estimator_spec(241 estimator_spec = _get_train_estimator_spec(245 return estimator_spec
216 estimator_spec = model_fn_lib.EstimatorSpec(221 estimator_spec = _get_eval_estimator_spec(gan_model, gan_loss,231 estimator_spec = _get_train_estimator_spec(gan_model, gan_loss, gopt,234 return estimator_spec
1462 estimator_spec = self._verify_estimator_spec(1464 loss, train_op = estimator_spec.loss, estimator_spec.train_op1466 … if isinstance(estimator_spec, model_fn_lib._TPUEstimatorSpec): # pylint: disable=protected-access1467 captured_scaffold_fn.capture(estimator_spec.scaffold_fn)1471 captured_training_hooks.capture(estimator_spec.training_hooks)1489 … if (isinstance(estimator_spec, model_fn_lib._TPUEstimatorSpec) # pylint: disable=protected-access1490 and estimator_spec.host_call is not None):1491 host_call.record({'host_call': estimator_spec.host_call})1707 estimator_spec = self._model_fn(features=features, **kwargs)1709 … isinstance(estimator_spec, model_fn_lib._TPUEstimatorSpec)): # pylint: disable=protected-access[all …]
35 estimator_spec = estimator.model_fn(39 export_outputs = estimator_spec.export_outputs
1252 * Remove estimator_spec(mode) argument.