Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
Megatron-LM
Commits
18846a0a
"install_cuda.sh" did not exist on "4cd63deff3b3cdc923e151c4efdaa281fa42d668"
Commit
18846a0a
authored
Jan 13, 2022
by
Lawrence McAfee
Browse files
removed argument flag; also, converted torch.FloatTenstor -> torch.empty in previous commit
parent
1f387c2c
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
0 additions
and
3 deletions
+0
-3
megatron/arguments.py
megatron/arguments.py
+0
-3
No files found.
megatron/arguments.py
View file @
18846a0a
...
@@ -681,9 +681,6 @@ def _add_distributed_args(parser):
...
@@ -681,9 +681,6 @@ def _add_distributed_args(parser):
help
=
'Call torch.cuda.empty_cache() each iteration '
help
=
'Call torch.cuda.empty_cache() each iteration '
'(training and eval), to reduce fragmentation.'
'(training and eval), to reduce fragmentation.'
'0=off, 1=moderate, 2=aggressive.'
)
'0=off, 1=moderate, 2=aggressive.'
)
group
.
add_argument
(
'--deallocate-pipeline-outputs'
,
action
=
'store_true'
,
default
=
False
,
help
=
'If set, pipeline output tensors '
'are deallocated during the forward pass.'
)
return
parser
return
parser
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a 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