Skip to content

Commit 6d1dd0d

Browse files
committed
Update train.py
1 parent 190072b commit 6d1dd0d

File tree

1 file changed

+0
-23
lines changed

1 file changed

+0
-23
lines changed

src/11_seq2seq/train.py

Lines changed: 0 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -47,11 +47,6 @@ def define_argparser(is_continue=False):
4747
default=0,
4848
help='GPU ID to train. Currently, GPU parallel is not supported. -1 for CPU. Default=%(default)s'
4949
)
50-
p.add_argument(
51-
'--off_autocast',
52-
action='store_true',
53-
help='Turn-off Automatic Mixed Precision (AMP), which speed-up training.',
54-
)
5550

5651
p.add_argument(
5752
'--batch_size',
@@ -127,24 +122,6 @@ def define_argparser(is_continue=False):
127122
default=1e-3,
128123
help='Initial learning rate. Default=%(default)s',
129124
)
130-
p.add_argument(
131-
'--lr_step',
132-
type=int,
133-
default=0,
134-
help='Number of epochs for each learning rate decay. Default=%(default)s',
135-
)
136-
p.add_argument(
137-
'--lr_gamma',
138-
type=float,
139-
default=.5,
140-
help='Learning rate decay rate. Default=%(default)s',
141-
)
142-
p.add_argument(
143-
'--lr_decay_start',
144-
type=int,
145-
default=10,
146-
help='Learning rate decay start at. Default=%(default)s',
147-
)
148125

149126
config = p.parse_args()
150127

0 commit comments

Comments
 (0)