Warning, /AID2E-framework/examples/dtlz2/configurations/dtlz2_ax_slurm.yml is written in an unsupported language. File is not indexed.
0001 # Slurm trial and workflow-job submission; update setup_commands for your cluster.
0002 problem:
0003 name: "DTLZ2 Slurm Workflow Example"
0004 problem_type: "DTLZ2"
0005 output_location: "./output/ax_slurm"
0006 work_location: "./work/ax_slurm"
0007 design_parameters_file: "../dtlz2_design.params"
0008 objectives:
0009 - name: "f1"
0010 direction: "minimize"
0011 - name: "f2"
0012 direction: "minimize"
0013
0014 optimizer:
0015 name: "ax"
0016 type: "bayesian"
0017 parameters:
0018 initialization_strategy: "sobol"
0019 generator: "BOTORCH_MODULAR"
0020 generator_kwargs:
0021 surrogate_spec:
0022 model_configs:
0023 - botorch_model_class: "SingleTaskGP"
0024 botorch_acqf_class: "qLogNoisyExpectedHypervolumeImprovement"
0025 acquisition_options:
0026 prune_baseline: true
0027 generator_gen_kwargs:
0028 model_gen_options:
0029 optimizer_kwargs:
0030 sequential: false
0031 num_restarts: 10
0032 objective_thresholds:
0033 f1: 1.0
0034 f2: 1.0
0035 n_initial_samples: 20
0036 n_iterations: 100
0037 batch_size: 4
0038 seed: 42
0039
0040 scheduler:
0041 runner_type: "SlurmRunner"
0042 parameters:
0043 job_name_prefix: "aid2e_cfg"
0044 runtime_working_dir: "."
0045 setup_commands:
0046 - "module load miniforge3/24.9.2-0"
0047 - "conda activate env_AID2E"
0048 poll_interval: 2
0049
0050 workflows:
0051 workflows:
0052 - name: "dtlz2_slurm_eval"
0053 description: "Canonical global-scheduler Slurm workflow example"
0054 branches:
0055 - name: "main"
0056 stages:
0057 - name: "evaluate"
0058 jobs:
0059 - name: "compute_dtlz2"
0060 command: "python examples/dtlz2/dtlz2_evaluator.py"
0061 rule: >-
0062 {command}
0063 --x1 {payload[x1]}
0064 --x2 {payload[x2]}
0065 --x3 {payload[x3]}
0066 --x4 {payload[x4]}
0067 --x5 {payload[x5]}
0068 --x6 {payload[x6]}
0069 --x7 {payload[x7]}
0070 --x8 {payload[x8]}
0071 --x9 {payload[x9]}
0072 --x10 {payload[x10]}
0073 --output {payload[output_file]}
0074 --label {payload[label]}
0075 --sleep-sec {payload[sleep_sec]}
0076 --repeat {payload[repeat]}
0077 payload:
0078 x1: "{design_point[DTLZ2_variables.x1]}"
0079 x2: "{design_point[DTLZ2_variables.x2]}"
0080 x3: "{design_point[DTLZ2_variables.x3]}"
0081 x4: "{design_point[DTLZ2_variables.x4]}"
0082 x5: "{design_point[DTLZ2_variables.x5]}"
0083 x6: "{design_point[DTLZ2_variables.x6]}"
0084 x7: "{design_point[DTLZ2_variables.x7]}"
0085 x8: "{design_point[DTLZ2_variables.x8]}"
0086 x9: "{design_point[DTLZ2_variables.x9]}"
0087 x10: "{design_point[DTLZ2_variables.x10]}"
0088 output_file: "{output_dir}/objectives.json"
0089 label: "config-driven-slurm"
0090 sleep_sec: 0.3
0091 repeat: 2
0092 outputs:
0093 - path: "{payload[output_file]}"
0094 format: "json"
0095 parallelism:
0096 max_concurrent: 1
0097 retry_max: 1
0098 timeout_sec: 300
