Skip to content

Commit bcb7713

Browse files
committed
Merge branch 'automl_paper' of https://github.com/suzhoum/autogluon-bench into automl_paper
2 parents 28a8682 + b21c952 commit bcb7713

File tree

1 file changed

+13
-11
lines changed

1 file changed

+13
-11
lines changed

sample_configs/cloud_configs/bench_all.py

Lines changed: 13 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -3,13 +3,13 @@
33
seeds = []
44
for i in range(n_experiments):
55
seeds.append(random.randint(0, 100))
6-
print(seeds)
76

87
seeds = [22, 92, 54, 86, 41]
8+
99
config_paths = [
1010
"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",
1313
# "sample_configs/multimodal_cloud_text_configs.yaml",
1414
# "sample_configs/multimodal_cloud_text_fs_configs.yaml",
1515
# "sample_configs/multimodal_cloud_text_tabular_configs.yaml",
@@ -18,13 +18,14 @@
1818
]
1919
frameworks = [
2020
# "AutoGluon_best_master",
21+
# "ablation_base",
2122
# "ablation_greedy_soup",
2223
# "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",
2627
# "ablation_lr_decay",
27-
"autokeras_master",
28+
# "autokeras_master",
2829
# "torch_compile_best",
2930
# "AutoGluon_best_master",
3031
# "AutoGluon_high_master",
@@ -47,8 +48,8 @@
4748
5,
4849
10
4950
]
50-
module = "autokeras"
51-
# module = "multimodal"
51+
# module = "autokeras"
52+
module = "multimodal"
5253

5354
import yaml
5455
import os
@@ -58,6 +59,7 @@
5859
os.makedirs(config_root, exist_ok=True)
5960

6061
for seed in seeds:
62+
print("Seed: ", seed)
6163
for constraint in constraints:
6264
os.makedirs(f"{config_root}/{constraint}", exist_ok=True)
6365
for framework in frameworks:
@@ -75,11 +77,11 @@
7577
configs["module"] = module
7678
configs["seed"] = seed
7779
# configs["custom_dataloader"]["shot"] = shot
78-
# configs["benchmark_name"] = f"{configs['benchmark_name']}-{shot}"
80+
configs["benchmark_name"] = f"{configs['benchmark_name']}-{seed}"
7981
new_config_path = os.path.join(config_dir, os.path.basename(config_path))
8082
with open(new_config_path, "w") as new_f:
8183
yaml.dump(configs, new_f)
82-
84+
print("Running config: ", new_config_path)
8385
command = ["agbench", "run", new_config_path]
8486
subprocess.run(command)
8587

0 commit comments

Comments
 (0)