Skip to content

Commit fad31c2

Browse files
committed
Update train.py
1 parent 5cf8911 commit fad31c2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/11_seq2seq/train.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def define_argparser(is_continue=False):
5656
p.add_argument(
5757
'--batch_size',
5858
type=int,
59-
default=32,
59+
default=256,
6060
help='Mini batch size for gradient descent. Default=%(default)s'
6161
)
6262
p.add_argument(
@@ -118,7 +118,7 @@ def define_argparser(is_continue=False):
118118
p.add_argument(
119119
'--iteration_per_update',
120120
type=int,
121-
default=8,
121+
default=1,
122122
help='Number of feed-forward iterations for one parameter update. Default=%(default)s'
123123
)
124124
p.add_argument(

0 commit comments

Comments
 (0)