DDP-Tutorial
DDP-Tutorial copied to clipboard
ENV LOCAL_RANK
你好,请问在env模式中,是如何获得arg.local_rank这个参数的。在启动方式中并没有看到。Tutorial中说可以通过环境变量自动获取,难道不需要添加这一条代码吗?gpu_id = int(os.environ['LOCAL_RANK'])