Skip to content

GitLab

Commit d4bb6cb0 authored by Boris Fomitchev's avatar Boris Fomitchev
Browse files

Making onnx-safe option overrideable 


Signed-off-by: default avatarBoris Fomitchev <bfomitchev@nvidia.com>
parent 3d1cbecf
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
megatron/arguments.py
View file @ d4bb6cb0
...@@ -169,7 +169,7 @@ def _add_network_size_args(parser): ...@@ -169,7 +169,7 @@ def _add_network_size_args(parser):
help='Use OpenAIs GeLU implementation. This option' help='Use OpenAIs GeLU implementation. This option'
'should not be used unless for backward compatibility' 'should not be used unless for backward compatibility'
'reasons.') 'reasons.')
group.add_argument('--onnx-safe', action='store_true', group.add_argument('--onnx-safe', type=bool, required=False,
help='Use workarounds for known problems with Torch ONNX exporter') help='Use workarounds for known problems with Torch ONNX exporter')
return parser 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