small cleanup

This commit is contained in:
pranz24
2020-06-06 00:38:20 +05:30
parent a1e8d7319e
commit 1a3f379b79
+1 -1
View File
@@ -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)')