File tree Expand file tree Collapse file tree 1 file changed +0
-23
lines changed Expand file tree Collapse file tree 1 file changed +0
-23
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments