在Tensorforce中呈现OpenAI环境

时间:2017-12-27 23:31:35

标签: tensorflow reinforcement-learning openai-gym

我正在训练代理人使用Tensorforce在OpenAI中玩HalfCheetah-v1环境。我能够训练代理并获得奖励,但在模型完成运行后我无法实际渲染环境。我使用以下代码保存了模型:

modelDir=os.path.join(os.getcwd(), "models")
agent.save_model(directory=os.path.join(modelDir, "agent"))

这很好用。但是,当我想再次加载模型以呈现训练有素的代理时,我收到以下错误:

RuntimeError: Attempted to use a closed Session.

以下代码是我实际用于渲染环境的代码。

saved_agent = agent.restore_model(os.path.join(modelDir))
env = gym.make("HalfCheetah-v1")
s = env.reset()
done = False
while not done:
    env.render()
    action = saved_agent.act(s)
    s, r, done, _ = env.step(action)

我的tensorforce版本号是0.3.3,而我的tensorflow-gpu版本号是1.4.1。请让我知道如何解决此错误。

0 个答案:

没有答案