From 1a3f379b79257d70e5896d6bdb38c2838bb836ab Mon Sep 17 00:00:00 2001 From: pranz24 Date: Sat, 6 Jun 2020 00:38:20 +0530 Subject: [PATCH] small cleanup --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 562e3d9..a3047c5 100644 --- a/main.py +++ b/main.py @@ -9,7 +9,7 @@ from tensorboardX import SummaryWriter from replay_memory import ReplayMemory parser = argparse.ArgumentParser(description='PyTorch Soft Actor-Critic Args') -parser.add_argument('--env-name', default="HalfCheetahBulletEnv-v0", +parser.add_argument('--env-name', default="HalfCheetah-v2", help='Mujoco Gym environment (default: HalfCheetah-v2)') parser.add_argument('--policy', default="Gaussian", help='Policy Type: Gaussian | Deterministic (default: Gaussian)')