tdpeuter/2025ML-project-neural_compression
tdpeuter/2025ML-project-neural_compression
Archived
1
Fork 0
Code Issues Pull requests Releases 1 Activity Actions

feat: Device override cli arg

Browse source
This commit is contained in:
Tibo De Peuter 2025-12-11 16:23:57 +01:00
parent 97e84a97db
commit e81997f129
Signed by: tdpeuter
GPG key ID: 38297DE43F75FFE2
2 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
main.py
View file

@ -7,7 +7,7 @@ from src.utils import determine_device
def main(): def main():
args, print_help = parse_arguments() args, print_help = parse_arguments()
device = determine_device() device = args.device or determine_device()
print(f"Running on device: {device}...") print(f"Running on device: {device}...")
match args.mode: match args.mode:

1
src/args.py
View file

@ -10,6 +10,7 @@ def parse_arguments():
parser.add_argument("--verbose", "-v", action="store_true", required=False, parser.add_argument("--verbose", "-v", action="store_true", required=False,
help="Enable verbose mode") help="Enable verbose mode")
parser.add_argument("--results", type=str, required=True, help="path to save graphs to") 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 = ArgumentParser(add_help=False)
dataparser.add_argument("--data-root", type=str, required=False) dataparser.add_argument("--data-root", type=str, required=False)