frac-bio-embed/job.yaml at main · rcurrie/frac-bio-embed · GitHub

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
apiVersion: batch/v1
kind: Job
metadata:
  name: frac-bio-embed-train
spec:
  backoffLimit: 0
  template:
    spec:
      restartPolicy: Never
      containers:
        - name: train
          image: gitlab-registry.nrp-nautilus.io/rcurrie/frac-bio-embed
          env:
            - name: AWS_ACCESS_KEY_ID
              valueFrom:
                secretKeyRef:
                  name: s3-west-credentials
                  key: access-key-id
            - name: AWS_SECRET_ACCESS_KEY
              valueFrom:
                secretKeyRef:
                  name: s3-west-credentials
                  key: secret-access-key
            - name: AWS_ENDPOINT_URL
              valueFrom:
                secretKeyRef:
                  name: s3-west-credentials
                  key: endpoint
            - name: S3_BASE
              value: "s3://braingeneers/personal/rcurrie/frac-bio-embed"
          resources:
            requests:
              memory: 8Gi
              cpu: 8
              nvidia.com/gpu: 1
              ephemeral-storage: 64Gi
            limits:
              memory: 8Gi
              cpu: 8
              nvidia.com/gpu: 1
              ephemeral-storage: 64Gi
          args: ["--max-samples", "100000", "--epochs", "20"]