时间戳

首页 > 解决方案 > 在 tensorflow 中创建 MonitoredTrainingSession 会导致 InvalidArgumentError

python - 在 tensorflow 中创建 MonitoredTrainingSession 会导致 InvalidArgumentError

问题描述

用于启动的代码MonitoredTrainingSession导致InvalidArgumentError：

with tf.train.MonitoredTrainingSession(
    server.target,
    is_chief=is_learner,
    checkpoint_dir=FLAGS.logdir,
    save_checkpoint_secs=600,
    save_summaries_secs=30,
    log_step_count_steps=50000,
    hooks=[py_process.PyProcessHook()],
    config=config) as session:

这是回溯：

Traceback (most recent call last):
  File "/Users/hughalessi/miniconda3/envs/rl/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 1334, in _do_call
    return fn(*args)
  File "/Users/hughalessi/miniconda3/envs/rl/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 1317, in _run_fn
    self._extend_graph()
  File "/Users/hughalessi/miniconda3/envs/rl/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 1352, in _extend_graph
    tf_session.ExtendSession(self._session)
tensorflow.python.framework.errors_impl.InvalidArgumentError: Input 1 of node scan/while/Merge_5 was passed double from scan/while/NextIteration_5:0 incompatible with expected float.

它是一个使用为可扩展代理构建的自定义环境，谷歌实现了 IMPALA 架构。这是存储库的链接： https ://github.com/deepmind/scalable_agent

标签： pythonreinforcement-learning

解决方案

推荐阅读