Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in / Register
Toggle navigation
Menu
Open sidebar
wanggh
apex
Commits
c1378e6f
Unverified
Commit
c1378e6f
authored
Jul 16, 2021
by
X Wang
Committed by
GitHub
Jul 16, 2021
Browse files
local_rank fix (#1129)
* local_rank and install cuda version fix
parent
d06404fe
Changes
1
Hide whitespace changes
Inline
Side-by-side
examples/imagenet/main_amp.py
View file @
c1378e6f
...
...
@@ -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.'
)
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment