|
3 | 3 | seeds = [] |
4 | 4 | for i in range(n_experiments): |
5 | 5 | seeds.append(random.randint(0, 100)) |
6 | | -print(seeds) |
7 | 6 |
|
8 | 7 | seeds = [22, 92, 54, 86, 41] |
| 8 | + |
9 | 9 | config_paths = [ |
10 | 10 | "sample_configs/paper_text_tabular_local_configs.yaml", |
11 | | - # "sample_configs/paper_text_local_configs.yaml", |
12 | | - # "sample_configs/paper_image_local_configs.yaml", |
| 11 | + "sample_configs/paper_text_local_configs.yaml", |
| 12 | + "sample_configs/paper_image_local_configs.yaml", |
13 | 13 | # "sample_configs/multimodal_cloud_text_configs.yaml", |
14 | 14 | # "sample_configs/multimodal_cloud_text_fs_configs.yaml", |
15 | 15 | # "sample_configs/multimodal_cloud_text_tabular_configs.yaml", |
|
18 | 18 | ] |
19 | 19 | frameworks = [ |
20 | 20 | # "AutoGluon_best_master", |
| 21 | + # "ablation_base", |
21 | 22 | # "ablation_greedy_soup", |
22 | 23 | # "ablation_gradient_clip", |
23 | | - # "ablation_warmup_steps", |
24 | | - # "ablation_cosine_decay", |
25 | | - # "ablation_weight_decay", |
| 24 | + "ablation_warmup_steps", |
| 25 | + "ablation_cosine_decay", |
| 26 | + "ablation_weight_decay", |
26 | 27 | # "ablation_lr_decay", |
27 | | - "autokeras_master", |
| 28 | + # "autokeras_master", |
28 | 29 | # "torch_compile_best", |
29 | 30 | # "AutoGluon_best_master", |
30 | 31 | # "AutoGluon_high_master", |
|
47 | 48 | 5, |
48 | 49 | 10 |
49 | 50 | ] |
50 | | -module = "autokeras" |
51 | | -# module = "multimodal" |
| 51 | +# module = "autokeras" |
| 52 | +module = "multimodal" |
52 | 53 |
|
53 | 54 | import yaml |
54 | 55 | import os |
|
58 | 59 | os.makedirs(config_root, exist_ok=True) |
59 | 60 |
|
60 | 61 | for seed in seeds: |
| 62 | + print("Seed: ", seed) |
61 | 63 | for constraint in constraints: |
62 | 64 | os.makedirs(f"{config_root}/{constraint}", exist_ok=True) |
63 | 65 | for framework in frameworks: |
|
75 | 77 | configs["module"] = module |
76 | 78 | configs["seed"] = seed |
77 | 79 | # configs["custom_dataloader"]["shot"] = shot |
78 | | - # configs["benchmark_name"] = f"{configs['benchmark_name']}-{shot}" |
| 80 | + configs["benchmark_name"] = f"{configs['benchmark_name']}-{seed}" |
79 | 81 | new_config_path = os.path.join(config_dir, os.path.basename(config_path)) |
80 | 82 | with open(new_config_path, "w") as new_f: |
81 | 83 | yaml.dump(configs, new_f) |
82 | | - |
| 84 | + print("Running config: ", new_config_path) |
83 | 85 | command = ["agbench", "run", new_config_path] |
84 | 86 | subprocess.run(command) |
85 | 87 |
|
0 commit comments