feat: first approach to action clipping
This commit is contained in:
parent
ac352ef431
commit
1ef257544a
5 changed files with 32 additions and 18 deletions
|
|
@ -69,18 +69,10 @@ if __name__ == "__main__":
|
|||
|
||||
env = make_env(args.env_config_path, args.num_envs)
|
||||
raw_env = env.raw
|
||||
print(
|
||||
"\n\n\n Observation space \n",
|
||||
raw_env.observation_space,
|
||||
"Action space \n",
|
||||
raw_env.action_space,
|
||||
)
|
||||
print(
|
||||
"\n\n\n Observation space \n",
|
||||
raw_env.observation_space,
|
||||
"Action space \n",
|
||||
raw_env.action_space,
|
||||
logger.log(
|
||||
{"run_dir": run_dir}
|
||||
)
|
||||
|
||||
torch.backends.cudnn.deterministic = args.torch_deterministic
|
||||
|
||||
ppo_trainer = PPOTrainer(args, env, run_dir, run_name)
|
||||
|
|
|
|||
Reference in a new issue