Skip to content

Commit

Permalink
Changes in reinforce.py (pytorch#140)
Browse files Browse the repository at this point in the history
  • Loading branch information
zuoxingdong authored and soumith committed Apr 11, 2017
1 parent 28471f2 commit 3f21078
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions reinforcement_learning/reinforce.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,10 +15,10 @@
parser.add_argument('--gamma', type=float, default=0.99, metavar='G',
help='discount factor (default: 0.99)')
parser.add_argument('--seed', type=int, default=543, metavar='N',
help='random seed (default: 1)')
help='random seed (default: 543)')
parser.add_argument('--render', action='store_true',
help='render the environment')
parser.add_argument('--log-interval', type=int, default=10, metavar='N',
parser.add_argument('--log_interval', type=int, default=10, metavar='N',
help='interval between training status logs (default: 10)')
args = parser.parse_args()

Expand Down Expand Up @@ -57,7 +57,6 @@ def select_action(state):

def finish_episode():
R = 0
saved_actions = model.saved_actions
rewards = []
for r in model.rewards[::-1]:
R = r + args.gamma * R
Expand Down

0 comments on commit 3f21078

Please sign in to comment.