feat: Device override cli arg

2025-12-11 16:23:57 +01:00 · 2025-12-11 16:23:57 +01:00 · e81997f129
commit e81997f129
parent 97e84a97db
2 changed files with 2 additions and 1 deletions
--- a/src/args.py
+++ b/src/args.py
@ -10,6 +10,7 @@ def parse_arguments():
    parser.add_argument("--verbose", "-v", action="store_true", required=False,
                        help="Enable verbose mode")
    parser.add_argument("--results", type=str, required=True, help="path to save graphs to")
+    parser.add_argument("--device", required=False, help="Override the device to use")

    dataparser = ArgumentParser(add_help=False)
    dataparser.add_argument("--data-root", type=str, required=False)