From ec38bbb32c3f72094e88ea02aef26d755a8fb99d Mon Sep 17 00:00:00 2001 From: Pranjal Tandon Date: Sun, 20 Jan 2019 15:46:01 +0530 Subject: [PATCH] Update main.py --- main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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',