summaryrefslogtreecommitdiff
path: root/cnn_v3/training/gen_test_vectors.py
diff options
context:
space:
mode:
authorskal <pascal.massimino@gmail.com>2026-03-26 07:10:19 +0100
committerskal <pascal.massimino@gmail.com>2026-03-26 07:10:19 +0100
commit130d70ea130242533a0fd3a7ffeabdb68598c88e (patch)
treec5f89a8cf6f9695b1799bb41c18a4bab7a2e9ad2 /cnn_v3/training/gen_test_vectors.py
parent8f14bdd66cb002b2f89265b2a578ad93249089c9 (diff)
fix(cnn_v3/training): fix defaults and help strings across py tools
- train_cnn_v3.py: enc-channels 4,8→8,16; checkpoint-every 50→100; add help strings for epochs/batch-size/lr/checkpoint-dir - gen_test_vectors.py: add help strings for --W/--H/--seed - export_cnn_v3_weights.py: fix --output help string (export/→export)
Diffstat (limited to 'cnn_v3/training/gen_test_vectors.py')
-rw-r--r--cnn_v3/training/gen_test_vectors.py9
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