summaryrefslogtreecommitdiff
path: root/cnn_v3/training/infer_cnn_v3.py
diff options
context:
space:
mode:
Diffstat (limited to 'cnn_v3/training/infer_cnn_v3.py')
-rw-r--r--cnn_v3/training/infer_cnn_v3.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/cnn_v3/training/infer_cnn_v3.py b/cnn_v3/training/infer_cnn_v3.py
index ca1c72a..b0fe9e6 100644
--- a/cnn_v3/training/infer_cnn_v3.py
+++ b/cnn_v3/training/infer_cnn_v3.py
@@ -129,8 +129,8 @@ def main():
p.add_argument('output', help='Output PNG')
p.add_argument('--checkpoint', '-c', metavar='CKPT',
help='Path to .pth checkpoint (auto-finds latest if omitted)')
- p.add_argument('--enc-channels', default='4,8',
- help='Encoder channels (default: 4,8 — must match checkpoint)')
+ p.add_argument('--enc-channels', default='8,16',
+ help='Encoder channels (default: 8,16 — must match checkpoint)')
p.add_argument('--cond', nargs=5, type=float, metavar='F', default=[0.0]*5,
help='FiLM conditioning: 5 floats (beat_phase beat_norm audio style0 style1)')
p.add_argument('--identity-film', action='store_true',