diff --git a/a2c_sb3_panda_reach.py b/a2c_sb3_panda_reach.py index a2cae910b3056dcb7c514e6964cd6eb742b144f5..ca4de874c619f48f494878d2db70eec85dc75223 100644 --- a/a2c_sb3_panda_reach.py +++ b/a2c_sb3_panda_reach.py @@ -1,7 +1,9 @@ import gymnasium as gym + from stable_baselines3 import A2C from stable_baselines3.common.monitor import Monitor from stable_baselines3.common.vec_env import DummyVecEnv + import wandb import panda_gym from wandb.integration.sb3 import WandbCallback