gaspersic/baselines
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

using mujoco_arg_parser as args

Browse Source 
remove origin parser
This commit is contained in:
20chase
2018-01-29 16:52:01 +08:00
committed by GitHub
parent 452b548c2a
commit 4a77855529
1 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
baselines/trpo_mpi/run_mujoco.py
View File

@@ -27,16 +27,10 @@ def train(env_id, num_timesteps, seed):
env.close() env.close()
def main(): def main():
import argparse
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--env', help='environment ID', default='Hopper-v1')
parser.add_argument('--seed', help='RNG seed', type=int, default=0)
parser.add_argument('--num-timesteps', type=int, default=int(1e6))
args = parser.parse_args()
args = mujoco_arg_parser().parse_args() args = mujoco_arg_parser().parse_args()
train(args.env, num_timesteps=args.num_timesteps, seed=args.seed) train(args.env, num_timesteps=args.num_timesteps, seed=args.seed)
if __name__ == '__main__': if __name__ == '__main__':
main() main()