swim/bo-ptycho
1
0
Fork 0
mirror of https://github.com/c-sooyoung/bo-ptycho.git synced 2026-09-17 20:29:07 +09:00
Code Issues Packages Projects Releases Wiki Activity

multi-gpu batched bo

Browse Source
This commit is contained in:
swim
2026-08-12 22:07:45 +09:00
parent 9c66bc84ad
commit 1c85c324c1
6 changed files with 253 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pipelines/sobo.py
+39 -37
View File
@@ -1,47 +1,49 @@
import os
import bo
import ptycho
# depreacated, use pipelines.batched_sobo.sobo_pipeline()
def sobo_pipeline(config):
# import os
# import bo
# import ptycho
result_dir = config['io']['result_dir']
os.makedirs(result_dir, exist_ok=True)
# def sobo_pipeline(config):
RANDOM_ITERS = config['job'].get('random_iters', 0)
SOBO_ITERS = config['job'].get('sobo_iters')
METRIC = config['bo']['metric']
PTYCHOENGINE = ptycho.engines[config['ptycho']['engine']]
# result_dir = config['io']['result_dir']
# os.makedirs(result_dir, exist_ok=True)
# RANDOM_ITERS = config['job'].get('random_iters', 0)
# SOBO_ITERS = config['job'].get('sobo_iters')
# METRIC = config['bo']['metric']
# PTYCHOENGINE = ptycho.engines[config['ptycho']['engine']]
randombo = bo.RandomBOEngine(config)
# randombo = bo.RandomBOEngine(config)
bo_txt = os.path.join(result_dir, "bo.txt")
with open(bo_txt, "w") as f:
f.write(f" iter\tmetric\t{"\t".join([p[:7] for p in randombo.params])}\n")
# bo_txt = os.path.join(result_dir, "bo.txt")
# with open(bo_txt, "w") as f:
# f.write(f" iter\tmetric\t{"\t".join([p[:7] for p in randombo.params])}\n")
for j in range(RANDOM_ITERS):
print(f"RANDOM sampling; iteration {j}")
job_config = randombo.ask()
ptycho_engine = PTYCHOENGINE(job_config)
ptycho_engine.run(run_id=f"bo-{j:03d}")
y_value = ptycho_engine.metric(METRIC)
randombo.tell(job_config, y_value)
with open(bo_txt, "a") as f:
p = [f'{job_config['ptycho']['params'][key]:.2f}' for key in randombo.params]
f.write(f"{j: 8d}\t{y_value:.4f}\t{"\t".join(p)}\n")
# for j in range(RANDOM_ITERS):
# print(f"RANDOM sampling; iteration {j}")
# job_config = randombo.ask()
# ptycho_engine = PTYCHOENGINE(job_config)
# ptycho_engine.run(run_id=f"bo-{j:03d}")
# y_value = ptycho_engine.metric(METRIC)
# randombo.tell(job_config, y_value)
# with open(bo_txt, "a") as f:
# p = [f'{job_config['ptycho']['params'][key]:.2f}' for key in randombo.params]
# f.write(f"{j: 8d}\t{y_value:.4f}\t{"\t".join(p)}\n")
sobo = bo.SingleObjectiveBOEngine(config)
sobo.train_x = randombo.train_x
sobo.train_y = randombo.train_y
# sobo = bo.SingleObjectiveBOEngine(config)
# sobo.train_x = randombo.train_x
# sobo.train_y = randombo.train_y
for j in range(SOBO_ITERS):
print(f"SOBO sampling; iteration {RANDOM_ITERS + j}")
job_config = sobo.ask()
ptycho_engine = PTYCHOENGINE(job_config)
ptycho_engine.run(run_id=f"bo-{RANDOM_ITERS + j:03d}")
y_value = ptycho_engine.metric(METRIC)
sobo.tell(job_config, y_value)
with open(bo_txt, "a") as f:
p = [f'{job_config['ptycho']['params'][key]:.2f}' for key in sobo.params]
f.write(f"{RANDOM_ITERS + j: 8d}\t{y_value:.4f}\t{"\t".join(p)}\n")
# for j in range(SOBO_ITERS):
# print(f"SOBO sampling; iteration {RANDOM_ITERS + j}")
# job_config = sobo.ask()
# ptycho_engine = PTYCHOENGINE(job_config)
# ptycho_engine.run(run_id=f"bo-{RANDOM_ITERS + j:03d}")
# y_value = ptycho_engine.metric(METRIC)
# sobo.tell(job_config, y_value)
# with open(bo_txt, "a") as f:
# p = [f'{job_config['ptycho']['params'][key]:.2f}' for key in sobo.params]
# f.write(f"{RANDOM_ITERS + j: 8d}\t{y_value:.4f}\t{"\t".join(p)}\n")