Commit 18846a0a authored by Lawrence McAfee's avatar Lawrence McAfee
Browse files

removed argument flag; also, converted torch.FloatTenstor -> torch.empty in previous commit

parent 1f387c2c
......@@ -681,9 +681,6 @@ def _add_distributed_args(parser):
help='Call torch.cuda.empty_cache() each iteration '
'(training and eval), to reduce fragmentation.'
'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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment