关于Tf.estimator.BaselineClassifier报错问题解决
-
我在执行如下代码时候报错：
baseline_estimator = tf.compat.v1.estimator.BaselineClassifier( model_dir = output_dir,n_classes = 2) baseline_estimator.train(input_fn = lambda: make_dataset( train_df, y_train,epochs=100)) -
报错内容
INFO:tensorflow:Using default config. INFO:tensorflow:Using config: {'_model_dir': 'baseline_model', '_tf_random_seed': None, '_save_summary_steps': 100, '_save_checkpoints_steps': None, '_save_checkpoints_secs': 600, '_session_config': allow_soft_placement: true graph_options { rewrite_options { meta_optimizer_iterations: ONE } } , '_keep_checkpoint_max': 5, '_keep_checkpoint_every_n_hours': 10000, '_log_step_count_steps': 100, '_train_distribute': None, '_device_fn': None, '_protocol': None, '_eval_distribute': None, '_experimental_distribute': None, '_experimental_max_worker_delay_secs': None, '_session_creation_timeout_secs': 7200, '_service': None, '_cluster_spec': ClusterSpec({}), '_task_type': 'worker', '_task_id': 0, '_global_id_in_cluster': 0, '_master': '', '_evaluation_master': '', '_is_chief': True, '_num_ps_replicas': 0, '_num_worker_replicas': 1} INFO:tensorflow:Calling model_fn. INFO:tensorflow:Done calling model_fn. INFO:tensorflow:Create CheckpointSaverHook. INFO:tensorflow:Graph was finalized. INFO:tensorflow:Restoring parameters from baseline_model/model.ckpt-1960 --------------------------------------------------------------------------- NotFoundError Traceback (most recent call last) /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/client/session.py in _do_call(self, fn, *args) 1366 try: -> 1367 return fn(*args) 1368 except errors.OpError as e: /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/client/session.py in _run_fn(feed_dict, fetch_list, target_list, options, run_metadata) 1351 return self._call_tf_sessionrun(options, feed_dict, fetch_list, -> 1352 target_list, run_metadata) 1353 /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/client/session.py in _call_tf_sessionrun(self, options, feed_dict, fetch_list, target_list, run_metadata) 1444 fetch_list, target_list, -> 1445 run_metadata) 1446 NotFoundError: Key training/Ftrl/baseline/bias/accumulator not found in checkpoint [[{{node save/RestoreV2}}]] During handling of the above exception, another exception occurred: NotFoundError Traceback (most recent call last) /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/saver.py in restore(self, sess, save_path) 1289 sess.run(self.saver_def.restore_op_name, -> 1290 {self.saver_def.filename_tensor_name: save_path}) 1291 except errors.NotFoundError as err: /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/client/session.py in run(self, fetches, feed_dict, options, run_metadata) 959 result = self._run(None, fetches, feed_dict, options_ptr, --> 960 run_metadata_ptr) 961 if run_metadata: /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/client/session.py in _run(self, handle, fetches, feed_dict, options, run_metadata) 1182 results = self._do_run(handle, final_targets, final_fetches, -> 1183 feed_dict_tensor, options, run_metadata) 1184 else: /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/client/session.py in _do_run(self, handle, target_list, fetch_list, feed_dict, options, run_metadata) 1360 return self._do_call(_run_fn, feeds, fetches, targets, options, -> 1361 run_metadata) 1362 else: /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/client/session.py in _do_call(self, fn, *args) 1385 'disable_meta_optimizer = True') -> 1386 raise type(e)(node_def, op, message) 1387 NotFoundError: Key training/Ftrl/baseline/bias/accumulator not found in checkpoint [[node save/RestoreV2 (defined at /opt/machine_learn/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/estimator.py:1493) ]] Original stack trace for 'save/RestoreV2': File "/opt/python3/lib/python3.7/runpy.py", line 193, in _run_module_as_main "__main__", mod_spec) File "/opt/python3/lib/python3.7/runpy.py", line 85, in _run_code exec(code, run_globals) File "/opt/machine_learn/lib/python3.7/site-packages/ipykernel_launcher.py", line 16, in <module> app.launch_new_instance() File "/opt/machine_learn/lib/python3.7/site-packages/traitlets/config/application.py", line 845, in launch_instance app.start() File "/opt/machine_learn/lib/python3.7/site-packages/ipykernel/kernelapp.py", line 612, in start self.io_loop.start() File "/opt/machine_learn/lib/python3.7/site-packages/tornado/platform/asyncio.py", line 199, in start self.asyncio_loop.run_forever() File "/opt/python3/lib/python3.7/asyncio/base_events.py", line 534, in run_forever self._run_once() File "/opt/python3/lib/python3.7/asyncio/base_events.py", line 1771, in _run_once handle._run() File "/opt/python3/lib/python3.7/asyncio/events.py", line 88, in _run self._context.run(self._callback, *self._args) File "/opt/machine_learn/lib/python3.7/site-packages/tornado/ioloop.py", line 688, in <lambda> lambda f: self._run_callback(functools.partial(callback, future)) File "/opt/machine_learn/lib/python3.7/site-packages/tornado/ioloop.py", line 741, in _run_callback ret = callback() File "/opt/machine_learn/lib/python3.7/site-packages/tornado/gen.py", line 814, in inner self.ctx_run(self.run) File "/opt/machine_learn/lib/python3.7/site-packages/tornado/gen.py", line 775, in run yielded = self.gen.send(value) File "/opt/machine_learn/lib/python3.7/site-packages/ipykernel/kernelbase.py", line 362, in process_one yield gen.maybe_future(dispatch(*args)) File "/opt/machine_learn/lib/python3.7/site-packages/tornado/gen.py", line 234, in wrapper yielded = ctx_run(next, result) File "/opt/machine_learn/lib/python3.7/site-packages/ipykernel/kernelbase.py", line 265, in dispatch_shell yield gen.maybe_future(handler(stream, idents, msg)) File "/opt/machine_learn/lib/python3.7/site-packages/tornado/gen.py", line 234, in wrapper yielded = ctx_run(next, result) File "/opt/machine_learn/lib/python3.7/site-packages/ipykernel/kernelbase.py", line 542, in execute_request user_expressions, allow_stdin, File "/opt/machine_learn/lib/python3.7/site-packages/tornado/gen.py", line 234, in wrapper yielded = ctx_run(next, result) File "/opt/machine_learn/lib/python3.7/site-packages/ipykernel/ipkernel.py", line 302, in do_execute res = shell.run_cell(code, store_history=store_history, silent=silent) File "/opt/machine_learn/lib/python3.7/site-packages/ipykernel/zmqshell.py", line 539, in run_cell return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs) File "/opt/machine_learn/lib/python3.7/site-packages/IPython/core/interactiveshell.py", line 2878, in run_cell raw_cell, store_history, silent, shell_futures) File "/opt/machine_learn/lib/python3.7/site-packages/IPython/core/interactiveshell.py", line 2923, in _run_cell return runner(coro) File "/opt/machine_learn/lib/python3.7/site-packages/IPython/core/async_helpers.py", line 68, in _pseudo_sync_runner coro.send(None) File "/opt/machine_learn/lib/python3.7/site-packages/IPython/core/interactiveshell.py", line 3147, in run_cell_async interactivity=interactivity, compiler=compiler, result=result) File "/opt/machine_learn/lib/python3.7/site-packages/IPython/core/interactiveshell.py", line 3338, in run_ast_nodes if (await self.run_code(code, result, async_=asy)): File "/opt/machine_learn/lib/python3.7/site-packages/IPython/core/interactiveshell.py", line 3418, in run_code exec(code_obj, self.user_global_ns, self.user_ns) File "<ipython-input-20-596fe090aad2>", line 8, in <module> baseline_estimator.train(input_fn = lambda: make_dataset( File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/estimator.py", line 374, in train loss = self._train_model(input_fn, hooks, saving_listeners) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/estimator.py", line 1164, in _train_model return self._train_model_default(input_fn, hooks, saving_listeners) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/estimator.py", line 1198, in _train_model_default saving_listeners) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/estimator.py", line 1493, in _train_with_estimator_spec log_step_count_steps=log_step_count_steps) as mon_sess: File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py", line 604, in MonitoredTrainingSession stop_grace_period_secs=stop_grace_period_secs) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py", line 1038, in __init__ stop_grace_period_secs=stop_grace_period_secs) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py", line 749, in __init__ self._sess = _RecoverableSession(self._coordinated_creator) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py", line 1231, in __init__ _WrappedSession.__init__(self, self._create_session()) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py", line 1236, in _create_session return self._sess_creator.create_session() File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py", line 902, in create_session self.tf_sess = self._session_creator.create_session() File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py", line 660, in create_session self._scaffold.finalize() File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py", line 243, in finalize self._saver.build() File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/saver.py", line 840, in build self._build(self._filename, build_save=True, build_restore=True) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/saver.py", line 878, in _build build_restore=build_restore) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/saver.py", line 502, in _build_internal restore_sequentially, reshape) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/saver.py", line 381, in _AddShardedRestoreOps name="restore_shard")) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/saver.py", line 328, in _AddRestoreOps restore_sequentially) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/saver.py", line 575, in bulk_restore return io_ops.restore_v2(filename_tensor, names, slices, dtypes) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/ops/gen_io_ops.py", line 1506, in restore_v2 name=name) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/framework/op_def_library.py", line 742, in _apply_op_helper attrs=attr_protos, op_def=op_def) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/framework/ops.py", line 3322, in _create_op_internal op_def=op_def) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/framework/ops.py", line 1756, in __init__ self._traceback = tf_stack.extract_stack() During handling of the above exception, another exception occurred: RuntimeError Traceback (most recent call last) /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/py_checkpoint_reader.py in get_tensor(self, tensor_str) 69 return CheckpointReader.CheckpointReader_GetTensor( ---> 70 self, compat.as_bytes(tensor_str)) 71 # TODO(b/143319754): Remove the RuntimeError casting logic once we resolve the RuntimeError: Key _CHECKPOINTABLE_OBJECT_GRAPH not found in checkpoint During handling of the above exception, another exception occurred: NotFoundError Traceback (most recent call last) /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/saver.py in restore(self, sess, save_path) 1299 try: -> 1300 names_to_keys = object_graph_key_mapping(save_path) 1301 except errors.NotFoundError: /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/saver.py in object_graph_key_mapping(checkpoint_path) 1617 reader = py_checkpoint_reader.NewCheckpointReader(checkpoint_path) -> 1618 object_graph_string = reader.get_tensor(trackable.OBJECT_GRAPH_PROTO_KEY) 1619 object_graph_proto = (trackable_object_graph_pb2.TrackableObjectGraph()) /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/py_checkpoint_reader.py in get_tensor(self, tensor_str) 73 except RuntimeError as e: ---> 74 error_translator(e) 75 /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/py_checkpoint_reader.py in error_translator(e) 34 'matching files for') in error_message: ---> 35 raise errors_impl.NotFoundError(None, None, error_message) 36 elif 'Sliced checkpoints are not supported' in error_message or ( NotFoundError: Key _CHECKPOINTABLE_OBJECT_GRAPH not found in checkpoint During handling of the above exception, another exception occurred: NotFoundError Traceback (most recent call last) <ipython-input-20-596fe090aad2> in <module> 6 baseline_estimator = tf.estimator.BaselineClassifier( 7 model_dir = output_dir,n_classes = 2) ----> 8 baseline_estimator.train(input_fn = lambda: make_dataset( 9 train_df, y_train,epochs=100)) /opt/machine_learn/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/estimator.py in train(self, input_fn, hooks, steps, max_steps, saving_listeners) 372 373 saving_listeners = _check_listeners_type(saving_listeners) --> 374 loss = self._train_model(input_fn, hooks, saving_listeners) 375 logging.info('Loss for final step: %s.', loss) 376 return self /opt/machine_learn/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/estimator.py in _train_model(self, input_fn, hooks, saving_listeners) 1162 return self._train_model_distributed(input_fn, hooks, saving_listeners) 1163 else: -> 1164 return self._train_model_default(input_fn, hooks, saving_listeners) 1165 1166 def _train_model_default(self, input_fn, hooks, saving_listeners): /opt/machine_learn/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/estimator.py in _train_model_default(self, input_fn, hooks, saving_listeners) 1196 return self._train_with_estimator_spec(estimator_spec, worker_hooks, 1197 hooks, global_step_tensor, -> 1198 saving_listeners) 1199 1200 def _train_model_distributed(self, input_fn, hooks, saving_listeners): /opt/machine_learn/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/estimator.py in _train_with_estimator_spec(self, estimator_spec, worker_hooks, hooks, global_step_tensor, saving_listeners) 1491 config=self._session_config, 1492 max_wait_secs=self._config.session_creation_timeout_secs, -> 1493 log_step_count_steps=log_step_count_steps) as mon_sess: 1494 loss = None 1495 any_step_done = False /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py in MonitoredTrainingSession(master, is_chief, checkpoint_dir, scaffold, hooks, chief_only_hooks, save_checkpoint_secs, save_summaries_steps, save_summaries_secs, config, stop_grace_period_secs, log_step_count_steps, max_wait_secs, save_checkpoint_steps, summary_dir, save_graph_def) 602 session_creator=session_creator, 603 hooks=all_hooks, --> 604 stop_grace_period_secs=stop_grace_period_secs) 605 606 /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py in __init__(self, session_creator, hooks, stop_grace_period_secs) 1036 hooks, 1037 should_recover=True, -> 1038 stop_grace_period_secs=stop_grace_period_secs) 1039 1040 /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py in __init__(self, session_creator, hooks, should_recover, stop_grace_period_secs) 747 stop_grace_period_secs=stop_grace_period_secs) 748 if should_recover: --> 749 self._sess = _RecoverableSession(self._coordinated_creator) 750 else: 751 self._sess = self._coordinated_creator.create_session() /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py in __init__(self, sess_creator) 1229 """ 1230 self._sess_creator = sess_creator -> 1231 _WrappedSession.__init__(self, self._create_session()) 1232 1233 def _create_session(self): /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py in _create_session(self) 1234 while True: 1235 try: -> 1236 return self._sess_creator.create_session() 1237 except _PREEMPTION_ERRORS as e: 1238 logging.info( /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py in create_session(self) 900 """Creates a coordinated session.""" 901 # Keep the tf_sess for unit testing. --> 902 self.tf_sess = self._session_creator.create_session() 903 # We don't want coordinator to suppress any exception. 904 self.coord = coordinator.Coordinator(clean_stop_exception_types=[]) /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py in create_session(self) 667 init_op=self._scaffold.init_op, 668 init_feed_dict=self._scaffold.init_feed_dict, --> 669 init_fn=self._scaffold.init_fn) 670 671 /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/session_manager.py in prepare_session(self, master, init_op, saver, checkpoint_dir, checkpoint_filename_with_path, wait_for_checkpoint, max_wait_secs, config, init_feed_dict, init_fn) 292 wait_for_checkpoint=wait_for_checkpoint, 293 max_wait_secs=max_wait_secs, --> 294 config=config) 295 if not is_loaded_from_checkpoint: 296 if init_op is None and not init_fn and self._local_init_op is None: /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/session_manager.py in _restore_checkpoint(self, master, saver, checkpoint_dir, checkpoint_filename_with_path, wait_for_checkpoint, max_wait_secs, config) 222 223 # Loads the checkpoint. --> 224 saver.restore(sess, ckpt.model_checkpoint_path) 225 saver.recover_last_checkpoints(ckpt.all_model_checkpoint_paths) 226 return sess, True /opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/saver.py in restore(self, sess, save_path) 1304 # a helpful message (b/110263146) 1305 raise _wrap_restore_error_with_msg( -> 1306 err, "a Variable name or other graph key that is missing") 1307 1308 # This is an object-based checkpoint. We'll print a warning and then do NotFoundError: Restoring from checkpoint failed. This is most likely due to a Variable name or other graph key that is missing from the checkpoint. Please ensure that you have not altered the graph expected based on the checkpoint. Original error: Key training/Ftrl/baseline/bias/accumulator not found in checkpoint [[node save/RestoreV2 (defined at /opt/machine_learn/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/estimator.py:1493) ]] Original stack trace for 'save/RestoreV2': File "/opt/python3/lib/python3.7/runpy.py", line 193, in _run_module_as_main "__main__", mod_spec) File "/opt/python3/lib/python3.7/runpy.py", line 85, in _run_code exec(code, run_globals) File "/opt/machine_learn/lib/python3.7/site-packages/ipykernel_launcher.py", line 16, in <module> app.launch_new_instance() File "/opt/machine_learn/lib/python3.7/site-packages/traitlets/config/application.py", line 845, in launch_instance app.start() File "/opt/machine_learn/lib/python3.7/site-packages/ipykernel/kernelapp.py", line 612, in start self.io_loop.start() File "/opt/machine_learn/lib/python3.7/site-packages/tornado/platform/asyncio.py", line 199, in start self.asyncio_loop.run_forever() File "/opt/python3/lib/python3.7/asyncio/base_events.py", line 534, in run_forever self._run_once() File "/opt/python3/lib/python3.7/asyncio/base_events.py", line 1771, in _run_once handle._run() File "/opt/python3/lib/python3.7/asyncio/events.py", line 88, in _run self._context.run(self._callback, *self._args) File "/opt/machine_learn/lib/python3.7/site-packages/tornado/ioloop.py", line 688, in <lambda> lambda f: self._run_callback(functools.partial(callback, future)) File "/opt/machine_learn/lib/python3.7/site-packages/tornado/ioloop.py", line 741, in _run_callback ret = callback() File "/opt/machine_learn/lib/python3.7/site-packages/tornado/gen.py", line 814, in inner self.ctx_run(self.run) File "/opt/machine_learn/lib/python3.7/site-packages/tornado/gen.py", line 775, in run yielded = self.gen.send(value) File "/opt/machine_learn/lib/python3.7/site-packages/ipykernel/kernelbase.py", line 362, in process_one yield gen.maybe_future(dispatch(*args)) File "/opt/machine_learn/lib/python3.7/site-packages/tornado/gen.py", line 234, in wrapper yielded = ctx_run(next, result) File "/opt/machine_learn/lib/python3.7/site-packages/ipykernel/kernelbase.py", line 265, in dispatch_shell yield gen.maybe_future(handler(stream, idents, msg)) File "/opt/machine_learn/lib/python3.7/site-packages/tornado/gen.py", line 234, in wrapper yielded = ctx_run(next, result) File "/opt/machine_learn/lib/python3.7/site-packages/ipykernel/kernelbase.py", line 542, in execute_request user_expressions, allow_stdin, File "/opt/machine_learn/lib/python3.7/site-packages/tornado/gen.py", line 234, in wrapper yielded = ctx_run(next, result) File "/opt/machine_learn/lib/python3.7/site-packages/ipykernel/ipkernel.py", line 302, in do_execute res = shell.run_cell(code, store_history=store_history, silent=silent) File "/opt/machine_learn/lib/python3.7/site-packages/ipykernel/zmqshell.py", line 539, in run_cell return super(ZMQInteractiveShell, self).run_cell(*args, **kwargs) File "/opt/machine_learn/lib/python3.7/site-packages/IPython/core/interactiveshell.py", line 2878, in run_cell raw_cell, store_history, silent, shell_futures) File "/opt/machine_learn/lib/python3.7/site-packages/IPython/core/interactiveshell.py", line 2923, in _run_cell return runner(coro) File "/opt/machine_learn/lib/python3.7/site-packages/IPython/core/async_helpers.py", line 68, in _pseudo_sync_runner coro.send(None) File "/opt/machine_learn/lib/python3.7/site-packages/IPython/core/interactiveshell.py", line 3147, in run_cell_async interactivity=interactivity, compiler=compiler, result=result) File "/opt/machine_learn/lib/python3.7/site-packages/IPython/core/interactiveshell.py", line 3338, in run_ast_nodes if (await self.run_code(code, result, async_=asy)): File "/opt/machine_learn/lib/python3.7/site-packages/IPython/core/interactiveshell.py", line 3418, in run_code exec(code_obj, self.user_global_ns, self.user_ns) File "<ipython-input-20-596fe090aad2>", line 8, in <module> baseline_estimator.train(input_fn = lambda: make_dataset( File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/estimator.py", line 374, in train loss = self._train_model(input_fn, hooks, saving_listeners) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/estimator.py", line 1164, in _train_model return self._train_model_default(input_fn, hooks, saving_listeners) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/estimator.py", line 1198, in _train_model_default saving_listeners) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/estimator.py", line 1493, in _train_with_estimator_spec log_step_count_steps=log_step_count_steps) as mon_sess: File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py", line 604, in MonitoredTrainingSession stop_grace_period_secs=stop_grace_period_secs) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py", line 1038, in __init__ stop_grace_period_secs=stop_grace_period_secs) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py", line 749, in __init__ self._sess = _RecoverableSession(self._coordinated_creator) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py", line 1231, in __init__ _WrappedSession.__init__(self, self._create_session()) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py", line 1236, in _create_session return self._sess_creator.create_session() File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py", line 902, in create_session self.tf_sess = self._session_creator.create_session() File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py", line 660, in create_session self._scaffold.finalize() File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/monitored_session.py", line 243, in finalize self._saver.build() File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/saver.py", line 840, in build self._build(self._filename, build_save=True, build_restore=True) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/saver.py", line 878, in _build build_restore=build_restore) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/saver.py", line 502, in _build_internal restore_sequentially, reshape) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/saver.py", line 381, in _AddShardedRestoreOps name="restore_shard")) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/saver.py", line 328, in _AddRestoreOps restore_sequentially) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/training/saver.py", line 575, in bulk_restore return io_ops.restore_v2(filename_tensor, names, slices, dtypes) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/ops/gen_io_ops.py", line 1506, in restore_v2 name=name) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/framework/op_def_library.py", line 742, in _apply_op_helper attrs=attr_protos, op_def=op_def) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/framework/ops.py", line 3322, in _create_op_internal op_def=op_def) File "/opt/machine_learn/lib/python3.7/site-packages/tensorflow_core/python/framework/ops.py", line 1756, in __init__ self._traceback = tf_stack.extract_stack() -
我的版本是 Tensorflow2.1.0 教程中为 2.0.0 alpha0 ,解决方法
baseline_estimator = tf.compat.v1.estimator.BaselineClassifier(model_dir=output_dir,n_classes=2)