1

我正在训练代理使用 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。请让我知道我应该如何解决这个错误。

4

0 回答 0