from local

This commit is contained in:
dohe0342 2022-12-10 14:16:57 +09:00
parent e961b355ee
commit 8931f3f708
2 changed files with 1 additions and 1 deletions

View File

@ -1115,7 +1115,7 @@ def run(rank, world_size, args, wb=None):
"""
params = get_params()
params.update(vars(args))
params.model_warm_step *= params.accum_grads
params.warm_step *= params.accum_grads
fix_random_seed(params.seed)
if world_size > 1: