rgb

a2c_sb3_cartpole.py

@@ -20,7 +20,7 @@ run = wandb.init(
     save_code=True,
 )
 
-env = gym.make("CartPole-v1")
+env = gym.make("CartPole-v1", render_mode="rgb_array")
 
 model = A2C("MlpPolicy", env, verbose=1, tensorboard_log=f"runs/{run.id}")
 #model = A2C("MlpPolicy", env, )