charts/maskrcnn-optimized/templates/maskrcnn.yaml [1:46]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
---
apiVersion: scheduling.k8s.io/v1
kind: PriorityClass
metadata:
  name: training-priority
value: 1000
preemptionPolicy: PreemptLowerPriority 
globalDefault: false
description: "This is the priority class for training"
---
apiVersion: kubeflow.org/v1alpha1
kind: MPIJob
metadata:
  name: {{ .Values.maskrcnn.name }}
  namespace: {{ .Values.global.namespace }} 
  labels:
    app.kubernetes.io/name: {{ .Values.maskrcnn.name }}
    app.kubernetes.io/instance: {{ .Release.Name }}
    app.kubernetes.io/managed-by: {{ .Release.Service }}
spec:
  gpus: {{ .Values.maskrcnn.gpus }}
  template:
    spec:
      priorityClassName: "training-priority"
      restartPolicy: Never
      volumes:
      - name: {{ .Values.global.shared_fs }} 
        persistentVolumeClaim:
            claimName: {{ .Values.global.shared_pvc }}
      - name: ebs
        hostPath:
            path: /ebs
            type: DirectoryOrCreate
      containers:
      - name: {{ .Values.maskrcnn.name }} 
        env:
        - name: HOROVOD_CYCLE_TIME
          value: "{{ .Values.maskrcnn.horovod_cycle_time }}" 
        - name: HOROVOD_FUSION_THRESHOLD
          value: "{{ .Values.maskrcnn.horovod_fusion_threshold }}" 
        - name: NCCL_SOCKET_IFNAME
          value: "{{ .Values.maskrcnn.nccl_socket_ifname }}" 
        - name: NCCL_MIN_NRINGS
          value: "{{ .Values.maskrcnn.nccl_min_rings }}" 
        - name: NCCL_DEBUG
          value: "{{ .Values.maskrcnn.nccl_debug }}" 
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



charts/maskrcnn/templates/maskrcnn.yaml [1:46]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
---
apiVersion: scheduling.k8s.io/v1
kind: PriorityClass
metadata:
  name: training-priority
value: 1000
preemptionPolicy: PreemptLowerPriority 
globalDefault: false
description: "This is the priority class for training"
---
apiVersion: kubeflow.org/v1alpha1
kind: MPIJob
metadata:
  name: {{ .Values.maskrcnn.name }}
  namespace: {{ .Values.global.namespace }} 
  labels:
    app.kubernetes.io/name: {{ .Values.maskrcnn.name }}
    app.kubernetes.io/instance: {{ .Release.Name }}
    app.kubernetes.io/managed-by: {{ .Release.Service }}
spec:
  gpus: {{ .Values.maskrcnn.gpus }}
  template:
    spec:
      priorityClassName: "training-priority"
      restartPolicy: Never
      volumes:
      - name: {{ .Values.global.shared_fs }} 
        persistentVolumeClaim:
            claimName: {{ .Values.global.shared_pvc }}
      - name: ebs
        hostPath:
            path: /ebs
            type: DirectoryOrCreate
      containers:
      - name: {{ .Values.maskrcnn.name }} 
        env:
        - name: HOROVOD_CYCLE_TIME
          value: "{{ .Values.maskrcnn.horovod_cycle_time }}" 
        - name: HOROVOD_FUSION_THRESHOLD
          value: "{{ .Values.maskrcnn.horovod_fusion_threshold }}" 
        - name: NCCL_SOCKET_IFNAME
          value: "{{ .Values.maskrcnn.nccl_socket_ifname }}" 
        - name: NCCL_MIN_NRINGS
          value: "{{ .Values.maskrcnn.nccl_min_rings }}" 
        - name: NCCL_DEBUG
          value: "{{ .Values.maskrcnn.nccl_debug }}" 
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



