Commit 537e2d9c authored by zhuwenwen's avatar zhuwenwen
Browse files

allow to set --enforce-eager

parent 76621631
......@@ -564,7 +564,7 @@ class EngineArgs:
'This should be a JSON string that will be '
'parsed into a dictionary.')
parser.add_argument('--enforce-eager',
# action='store_true',
action='store_true',
default=True,
help='Always use eager-mode PyTorch. If False, '
'will use eager mode and CUDA graph in hybrid '
......
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