feat: Add model choice

2025-12-06 21:52:31 +01:00 · 2025-12-06 21:52:31 +01:00 · ef50d6321e
commit ef50d6321e
parent bb241154d9
10 changed files with 102 additions and 54 deletions
--- a/main.py
+++ b/main.py
@ -22,7 +22,9 @@ def main():
                data_root = args.data_root,
                n_trials = 3 if args.debug else None,
                size = 2**10 if args.debug else None,
-                model_path = args.model_path
+                model_name=args.model,
+                model_path = args.model_load_path,
+                model_out = args.model_save_path
            )

        case 'compress':