diff options
Diffstat (limited to 'cnn_v3/training/gen_test_vectors.py')
| -rw-r--r-- | cnn_v3/training/gen_test_vectors.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/cnn_v3/training/gen_test_vectors.py b/cnn_v3/training/gen_test_vectors.py index cdda5a5..3f81247 100644 --- a/cnn_v3/training/gen_test_vectors.py +++ b/cnn_v3/training/gen_test_vectors.py @@ -405,9 +405,12 @@ def main(): parser = argparse.ArgumentParser(description="CNN v3 parity test vector generator") parser.add_argument('--header', action='store_true', help='Emit C header to stdout') - parser.add_argument('--W', type=int, default=8) - parser.add_argument('--H', type=int, default=8) - parser.add_argument('--seed', type=int, default=42) + parser.add_argument('--W', type=int, default=8, + help='Test image width (default 8)') + parser.add_argument('--H', type=int, default=8, + help='Test image height (default 8)') + parser.add_argument('--seed', type=int, default=42, + help='Random seed (default 42)') args = parser.parse_args() # Send self-test output to stderr so --header stdout stays clean |