Repository navigation
Expand file tree
/
Copy path_ca1_bs.py
More file actions
51 lines (49 loc) · 2.33 KB
/
Copy path_ca1_bs.py
File metadata and controls
51 lines (49 loc) · 2.33 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
from machinable import Interface, get
class _Ca1Bs(Interface):
def launch(self):
nm = "BS"
for repeat in range(3):
with get(
"machinable.scope",
{"trial": 0, "repeat": repeat},
):
for backbone in [
"fttransformer",
# "resnet"
]:
for target in [
"objective constraint distance",
"objective distance",
"objective constraint",
"objective",
"constraint",
"distance",
]:
get(
"interface.sopt_modeling",
[
f"""~from_protocol("benchmarks/ca1_pinsky_rinzel_modeling/config/CA1_{nm}.yaml")""",
{
"dopt_params": {
"dynamic_initial_sampling": "models.ops.dynamic_sampling",
"dynamic_initial_sampling_kwargs": {
"samples_per_iteration": 100,
"max_samples": 1200,
"stop_condition": False,
"optimizer_sampling": None,
"feasibility_solving": True,
"feasibility_targets": target,
"backbone": backbone,
},
"n_initial": 10,
"surrogate_custom_training": "models.ops.joint",
"n_epochs": 25,
"surrogate_custom_training_kwargs": {
"mode": "c+o",
"sensitivity": True,
"backbone": backbone,
},
}
},
],
).launch()