Update main.py

This commit is contained in:
Pranjal Tandon
2019-04-05 01:12:31 +05:30
committed by GitHub
parent ee617f3fed
commit d273b0221c
+1 -1
View File
@@ -28,7 +28,7 @@ parser.add_argument('--automatic_entropy_tuning', type=bool, default=False, meta
help='Temperature parameter α automaically adjusted.')
parser.add_argument('--seed', type=int, default=456, metavar='N',
help='random seed (default: 456)')
parser.add_argument('--batch_size', type=int, default=100, metavar='N',
parser.add_argument('--batch_size', type=int, default=256, metavar='N',
help='batch size (default: 256)')
parser.add_argument('--num_steps', type=int, default=1000001, metavar='N',
help='maximum number of steps (default: 1000000)')