from local

This commit is contained in:
dohe0342 2023-02-03 01:17:28 +09:00
parent 1bc959a300
commit 81af9bbc6d
2 changed files with 1 additions and 2 deletions

View File

@ -216,7 +216,7 @@ def get_params() -> AttributeDict:
"weight_decay": 1e-5, "weight_decay": 1e-5,
"lr_factor": 3.0, "lr_factor": 3.0,
#"warm_step": 36000, #"warm_step": 36000,
"warm_step": 1000, "warm_step": 2000,
"env_info": get_env_info(), "env_info": get_env_info(),
} }
) )
@ -581,7 +581,6 @@ def run(rank, world_size, args):
params.update(vars(args)) params.update(vars(args))
fix_random_seed(params.seed) fix_random_seed(params.seed)
print(rank)
if world_size > 1: if world_size > 1:
setup_dist(rank, world_size, params.master_port) setup_dist(rank, world_size, params.master_port)