Skip to content

Commit 181ac16

Browse files
rarilureloapaszke
authored andcommitted
fix argparse type of gamma
1 parent d10a529 commit 181ac16

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

reinforcement_learning/actor_critic.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414

1515

1616
parser = argparse.ArgumentParser(description='PyTorch actor-critic example')
17-
parser.add_argument('--gamma', type=int, default=0.99, metavar='G',
17+
parser.add_argument('--gamma', type=float, default=0.99, metavar='G',
1818
help='discount factor (default: 0.99)')
1919
parser.add_argument('--seed', type=int, default=543, metavar='N',
2020
help='random seed (default: 1)')

reinforcement_learning/reinforce.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414

1515

1616
parser = argparse.ArgumentParser(description='PyTorch REINFORCE example')
17-
parser.add_argument('--gamma', type=int, default=0.99, metavar='G',
17+
parser.add_argument('--gamma', type=float, default=0.99, metavar='G',
1818
help='discount factor (default: 0.99)')
1919
parser.add_argument('--seed', type=int, default=543, metavar='N',
2020
help='random seed (default: 1)')

0 commit comments

Comments
 (0)