mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Change base lr to 0.1, also rename from initial lr in train.py
This commit is contained in:
parent
146626bb85
commit
11886dc4f6
@ -230,11 +230,10 @@ def get_parser():
|
||||
)
|
||||
|
||||
parser.add_argument(
|
||||
"--initial-lr",
|
||||
"--base-lr",
|
||||
type=float,
|
||||
default=0.075,
|
||||
help="The initial learning rate. This value should not need "
|
||||
"to be changed.",
|
||||
default=0.1,
|
||||
help="The base learning rate."
|
||||
)
|
||||
|
||||
parser.add_argument(
|
||||
@ -977,7 +976,7 @@ def run(rank, world_size, args):
|
||||
find_unused_parameters=True)
|
||||
|
||||
optimizer = ScaledAdam(model.parameters(),
|
||||
lr=params.initial_lr,
|
||||
lr=params.base_lr,
|
||||
clipping_scale=2.0)
|
||||
|
||||
scheduler = Eden(optimizer, params.lr_batches, params.lr_epochs)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user