Warning, /AID2E-framework/examples/basic/full_example_slurm.yml is written in an unsupported language. File is not indexed.
0001 # AID2E Full Example Configuration with SLURM HPC Scheduler
0002 # This example demonstrates using the SLURM runner for HPC cluster execution
0003
0004 problem:
0005 name: "DTLZ2 Multi-Objective Optimization on HPC"
0006 problem_type: "toy"
0007 output_location: "/tmp"
0008 work_location: "/tmp"
0009
0010 objectives:
0011 - name: "f1"
0012 direction: "minimize"
0013 computation:
0014 steps:
0015 stages:
0016 - name: "evaluate_f1"
0017 script:
0018 path: "scripts/dtlz2_problem.py"
0019 output_file: "objectives_{job_id}.json"
0020 timeout_sec: 600
0021 produces_objective: true
0022 metrics_keys: ["f1"]
0023 - name: "f2"
0024 direction: "minimize"
0025 computation:
0026 steps:
0027 stages:
0028 - name: "evaluate_f2"
0029 script:
0030 path: "scripts/dtlz2_problem.py"
0031 output_file: "objectives_{job_id}.json"
0032 timeout_sec: 600
0033 produces_objective: true
0034 metrics_keys: ["f2"]
0035
0036 design_parameters_file: "design.params"
0037
0038 optimizer:
0039 name: "ax"
0040 type: "bayesian"
0041 parameters:
0042 initialization_strategy: "sobol"
0043 generator: "BOTORCH_MODULAR"
0044 n_initial_samples: 24
0045 n_iterations: 100
0046 batch_size: 4
0047 seed: 42
0048
0049 # Scheduler configuration for HPC execution using SLURM
0050 scheduler:
0051 runner_type: "SlurmRunner"
0052 parameters:
0053 partition: "gpu" # SLURM queue/partition name
0054 ntasks: 1
0055 cpus_per_task: 8 # Use 8 CPUs per task
0056 mem: "32GB" # Allocate 32 GB memory per task
0057 time: "12:00:00" # 12 hour wall clock limit
0058 gres: "gpu:1" # Request 1 GPU per task
0059 job_name_prefix: "aid2e_dtlz2"
0060 constraint: "gpu:v100" # Specific GPU constraint
0061 max_retries: 3
0062 output_location: "./slurm_output"
0063 monitor_interval: 60 # Check job status every 60 seconds
0064
0065 metadata:
0066 project: "AID2E"
0067 description: "AI assisted Detector Design for EIC - HPC Execution"
0068 version: "0.0.1"
