---
apiVersion: config.kubeflow.org/v1beta1
kind: KatibConfig
init:
  certGenerator:
    enable: true
  controller:
    webhookPort: 8443
    enableLeaderElection: true
    trialResources:
      - Job.v1.batch
      - TFJob.v1.kubeflow.org
      - PyTorchJob.v1.kubeflow.org
      - MPIJob.v1.kubeflow.org
      - XGBoostJob.v1.kubeflow.org
runtime:
  metricsCollectors:
    - kind: StdOut
      image: ghcr.io/kubeflow/katib/file-metrics-collector:v0.18.0
    - kind: File
      image: ghcr.io/kubeflow/katib/file-metrics-collector:v0.18.0
    - kind: TensorFlowEvent
      image: ghcr.io/kubeflow/katib/tfevent-metrics-collector:v0.18.0
      resources:
        limits:
          memory: 1Gi
  suggestions:
    - algorithmName: random
      image: ghcr.io/kubeflow/katib/suggestion-hyperopt:v0.18.0
    - algorithmName: tpe
      image: ghcr.io/kubeflow/katib/suggestion-hyperopt:v0.18.0
    - algorithmName: grid
      image: ghcr.io/kubeflow/katib/suggestion-optuna:v0.18.0
    - algorithmName: hyperband
      image: ghcr.io/kubeflow/katib/suggestion-hyperband:v0.18.0
    - algorithmName: bayesianoptimization
      image: ghcr.io/kubeflow/katib/suggestion-skopt:v0.18.0
    - algorithmName: cmaes
      image: ghcr.io/kubeflow/katib/suggestion-goptuna:v0.18.0
    - algorithmName: sobol
      image: ghcr.io/kubeflow/katib/suggestion-goptuna:v0.18.0
    - algorithmName: multivariate-tpe
      image: ghcr.io/kubeflow/katib/suggestion-optuna:v0.18.0
    - algorithmName: enas
      image: ghcr.io/kubeflow/katib/suggestion-enas:v0.18.0
      resources:
        limits:
          memory: 400Mi
    - algorithmName: darts
      image: ghcr.io/kubeflow/katib/suggestion-darts:v0.18.0
    - algorithmName: pbt
      image: ghcr.io/kubeflow/katib/suggestion-pbt:v0.18.0
      persistentVolumeClaimSpec:
        accessModes:
          - ReadWriteMany
        resources:
          requests:
            storage: 5Gi
  earlyStoppings:
    - algorithmName: medianstop
      image: ghcr.io/kubeflow/katib/earlystopping-medianstop:v0.18.0
