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)')