Update main.py

This commit is contained in:
Pranjal Tandon
2018-12-05 02:14:35 +05:30
committed by GitHub
parent 5aa3f5d532
commit ffde1723f9
+1 -1
View File
@@ -25,7 +25,7 @@ parser.add_argument('--tau', type=float, default=0.005, metavar='G',
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.2, metavar='G',
help='Entropy regularization coefficient is equivalent to inverse of reward scale. It is much more stable for training. (default: 0.2)')
help='emperature parameter α determines the relative importance of the entropy term against the reward (default: 0.2)'
parser.add_argument('--seed', type=int, default=543, metavar='N',
help='random seed (default: 543)')
parser.add_argument('--batch_size', type=int, default=256, metavar='N',