diff --git a/main.py b/main.py index d96d203..1253cd1 100644 --- a/main.py +++ b/main.py @@ -20,8 +20,8 @@ parser.add_argument('--gamma', type=float, default=0.99, metavar='G', help='discount factor for reward (default: 0.99)') parser.add_argument('--tau', type=float, default=0.005, metavar='G', help='target smoothing coefficient(τ) (default: 0.005)') -parser.add_argument('--lr', type=float, default=0.001, metavar='G', - help='learning rate (default: 0.001)') +parser.add_argument('--lr', type=float, default=0.0003, metavar='G', + help='learning rate (default: 0.0003)') parser.add_argument('--alpha', type=float, default=0.1, metavar='G', help='Temperature parameter α determines the relative importance of the entropy term against the reward (default: 0.1)') parser.add_argument('--automatic_entropy_tuning', type=bool, default=False, metavar='G',