未验证 提交 21fa0346 编写于 作者: S sneaxiy 提交者: GitHub

fix launch unorder (#55010)

上级 23c24af9
......@@ -200,4 +200,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.
先完成此消息的编辑!
想要评论请 注册