()
| 1295 | f'Couldn\'t find trained model at {self._model_dir}.') |
| 1296 | |
| 1297 | def _fn(): |
| 1298 | random_seed.set_random_seed(self._config.tf_random_seed) |
| 1299 | |
| 1300 | input_receiver_fn = input_receiver_fn_map[mode] |
| 1301 | input_receiver = input_receiver_fn() |
| 1302 | estimator_spec = self._call_model_fn( |
| 1303 | features=input_receiver.features, |
| 1304 | labels=getattr(input_receiver, 'labels', None), |
| 1305 | mode=mode, |
| 1306 | config=self.config) |
| 1307 | export_outputs = export_lib.export_outputs_for_mode( |
| 1308 | mode=estimator_spec.mode, |
| 1309 | serving_export_outputs=estimator_spec.export_outputs, |
| 1310 | predictions=estimator_spec.predictions, |
| 1311 | loss=estimator_spec.loss, |
| 1312 | metrics=estimator_spec.eval_metric_ops) |
| 1313 | signature_def_map = export_lib.build_all_signature_defs( |
| 1314 | input_receiver.receiver_tensors, |
| 1315 | export_outputs, |
| 1316 | getattr(input_receiver, |
| 1317 | 'receiver_tensors_alternatives', None), |
| 1318 | serving_only=(mode == ModeKeys.PREDICT)) |
| 1319 | main_op = None |
| 1320 | if estimator_spec.scaffold.local_init_op is not None: |
| 1321 | main_op = estimator_spec.scaffold.local_init_op |
| 1322 | return signature_def_map, main_op |
| 1323 | |
| 1324 | return export_all( |
| 1325 | export_dir_base, |
no test coverage detected