fixup! feat: Size arg

2025-12-10 16:34:53 +01:00 · 2025-12-10 16:34:53 +01:00 · 64bbae6300
commit 64bbae6300
parent 67310ce4f4
1 changed files with 1 additions and 1 deletions
--- a/main.py
+++ b/main.py
@ -12,7 +12,7 @@ def main():

    match args.mode:
        case 'train':
-            size = args.size
+            size = int(args.size) if args.size else None
            if args.method == 'optuna':
                size = 2 ** 12
                print(f"Using size {size} for optuna (was {args.size})")