未验证 提交 8ede4a9c 编写于 作者: S sneaxiy 提交者: GitHub

fix launch unorder (#55011)

上级 bae4cbec
......@@ -222,4 +222,9 @@ def parse_args():
help="seconds to wait before elastic job begin to train",
)
return parser.parse_known_args()
args = parser.parse_known_args()
env_rank = int(os.getenv('PADDLE_TRAINER_ID', -1))
if env_rank >= 0:
assert hasattr(args[0], "rank")
args[0].rank = env_rank
return args
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册