deep-learning - OpenAI 健身房录制视频

Question

我想录制我创建的模型的预测视频。请帮忙：

我目前正在使用 OpenAI Gym 包装器中的 VideoRecorder。现在它只是给出了一个分段错误。

for i in range(200):
    action, _states = model.predict(obs)
    obs, reward, done, info = env.step(action)
    env.render()
    v.capture_frame()
    v.close()
    #print("Reward: {} for iteration: {}".format(reward, i))
    if done:
        print(count)
        count+=1
        obs = env.reset()