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

fix: Subset Optuna

Browse source
This commit is contained in:
Tibo De Peuter 2025-12-10 11:09:44 +01:00
parent 51e0ed7fc0
commit 956ff79fa1
Signed by: tdpeuter
GPG key ID: 38297DE43F75FFE2
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
main.py
View file

@ -12,12 +12,17 @@ def main():
match args.mode: match args.mode:
case 'train': case 'train':
size = None
if args.method == 'optuna':
size = 2 ** 12
if args.debug:
size = 2 ** 10
train( train(
device=device, device=device,
dataset=args.dataset, dataset=args.dataset,
data_root=args.data_root, data_root=args.data_root,
n_trials=3 if args.debug else None, n_trials=3 if args.debug else None,
size=2 ** 10 if args.debug else None, size=size,
method=args.method, method=args.method,
model_name=args.model, model_name=args.model,
model_path=args.model_load_path, model_path=args.model_load_path,