Increase warm_step (and valid_interval)

This commit is contained in:
Daniel Povey 2022-03-22 13:32:24 +08:00
parent b82a505dfc
commit 4004ca81d8

View File

@ -295,7 +295,7 @@ def get_params() -> AttributeDict:
# parameters for decoder
"embedding_dim": 512,
# parameters for Noam
"warm_step": 30000, # For the 100h subset, use 8k
"warm_step": 60000, # For the 100h subset, use 8k
"model_warm_step": 3000, # arg given to model, not for lrate
"env_info": get_env_info(),
}
@ -689,8 +689,8 @@ def run(rank, world_size, args):
params = get_params()
params.update(vars(args))
if params.full_libri is False:
params.valid_interval = 800
params.warm_step = 16000
params.valid_interval = 1600
params.warm_step = 30000
fix_random_seed(params.seed)
if world_size > 1: