Skip to content

Conversation

@zou3519
Copy link
Contributor

@zou3519 zou3519 commented Apr 9, 2018

Fixes #6312.

Changed bottleneck's arg parser to user argparse.REMAINDER. This lets
the user specify args as python -m torch.utils.bottleneck script.py [args] (previously, a -- was needed after bottleneck and before
script.py).

cc @fmassa

Test Plan

Unit test in the bottleneck test suite. The test script errors out if any of the arguments are not provided.

Fixes pytorch#6312.

Changed bottleneck's arg parser to user argparse.REMAINDER. This lets
the user specify args as `python -m torch.utils.bottleneck script.py
[args]` (previously, a -- was needed after `bottleneck` and before
`script.py`).
@soumith soumith merged commit 1b3a5a4 into pytorch:master Apr 9, 2018
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants