0

I am trying to implement a code for saving videos every 500 episodes while using OpenAI gym and training. I was unable to get videos every 500 episodes.

Code:

from gym import wrappers
import gym    
vidsavedir = "./video"
    vidsaveeachepi = 500
    
    env = gym.make("MsPacman-v0")
    env = wrappers.Monitor(env, vidsavedir, video_callable = lambda episode_id: (episode_id % vidsaveeachepi) == 0, force = True)
    env.reset()
    
    for i in range(1,2):
      for step in range(1000):
        env.render()
        
        action = env.action_space.sample()         
        observation, reward, done, info = env.step(action)    
            
        if done:
          env.reset()
          print(step) 
          break;
      print(i) 
     
    
    env.close()

When I print steps it shows more than 500 but the video does not save. Only one video can see in the folder.

How can I solve this issue?

Ind
  • 377
  • 2
  • 6
  • 16

0 Answers0