diff options
| author | skal <pascal.massimino@gmail.com> | 2026-03-26 07:10:19 +0100 |
|---|---|---|
| committer | skal <pascal.massimino@gmail.com> | 2026-03-26 07:10:19 +0100 |
| commit | 130d70ea130242533a0fd3a7ffeabdb68598c88e (patch) | |
| tree | c5f89a8cf6f9695b1799bb41c18a4bab7a2e9ad2 /cnn_v3/training/export_cnn_v3_weights.py | |
| parent | 8f14bdd66cb002b2f89265b2a578ad93249089c9 (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/export_cnn_v3_weights.py')
| -rw-r--r-- | cnn_v3/training/export_cnn_v3_weights.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/cnn_v3/training/export_cnn_v3_weights.py b/cnn_v3/training/export_cnn_v3_weights.py index 2fa83d1..67330a6 100644 --- a/cnn_v3/training/export_cnn_v3_weights.py +++ b/cnn_v3/training/export_cnn_v3_weights.py @@ -189,7 +189,7 @@ def main() -> None: p = argparse.ArgumentParser(description='Export CNN v3 trained weights to .bin') p.add_argument('checkpoint', help='Path to .pth checkpoint file') p.add_argument('--output', default='export', - help='Output directory (default: export/)') + help='Output directory (default: export)') p.add_argument('--html', action='store_true', help=f'Also update {_WEIGHTS_JS_DEFAULT} with base64-encoded weights') p.add_argument('--html-output', default=None, metavar='PATH', |