创建MonitoredTrainingSession会在张量流中导致InvalidArgumentError

时间:2018-12-17 23:45:02

标签: python reinforcement-learning

用于初始化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.

这是一个使用自定义环境的环境,该环境是为Google的IMPALA架构实现可伸缩代理构建的。这是存储库的链接: https://github.com/deepmind/scalable_agent

0 个答案:

没有答案