未验证 提交 c1378e6f 编辑于 作者: X Wang's avatar X Wang 提交者: GitHub
浏览文件

local_rank fix (#1129)

* local_rank and install cuda version fix
上级 d06404fe
......@@ -81,7 +81,7 @@ def parse():
help='Only run 10 iterations for profiling.')
parser.add_argument('--deterministic', action='store_true')
parser.add_argument("--local_rank", default=0, type=int)
parser.add_argument("--local_rank", default=os.getenv('LOCAL_RANK', 0), type=int)
parser.add_argument('--sync_bn', action='store_true',
help='enabling apex sync BN.')
......
支持 Markdown
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册