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

fixup! feat: Size arg

Browse source
This commit is contained in:
Tibo De Peuter 2025-12-10 16:34:53 +01:00
parent 67310ce4f4
commit 64bbae6300
Signed by: tdpeuter
GPG key ID: 38297DE43F75FFE2
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
main.py
View file

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