Bump disk for student (#786)

* Bump disk for student * Fix alias
mozilla · Aug 29, 2024 · 012a45c · 012a45c
1 parent 82efece
commit 012a45c
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/taskcluster/config.yml b/taskcluster/config.yml
@@ -181,6 +181,12 @@ local-worker-aliases:
         by-worker-class:
             gcp-standard: 'b-linux-v100-gpu-4-1tb-standard'
             default: 'b-linux-v100-gpu-4-1tb'
+    # Use for student training, see https://github.com/mozilla/firefox-translations-training/issues/774
+    b-largegpu-xxlargedisk:
+        by-worker-class:
+            gcp-standard: 'b-linux-v100-gpu-4-2tb'
+            default: 'b-linux-v100-gpu-4-2tb'
+
 
 # Keys are worker type, and align with the `worker-type` entries in the
 # `worker.aliases` above.

diff --git a/taskcluster/kinds/finetune-student/kind.yml b/taskcluster/kinds/finetune-student/kind.yml
@@ -59,7 +59,7 @@ tasks:
         # Don't run unless explicitly scheduled
         run-on-tasks-for: []
 
-        worker-type: b-largegpu-xlargedisk
+        worker-type: b-largegpu-xxlargedisk
         worker:
             max-run-time: 2592000
             # train_taskcluster.py exits with 17 if a request to Taskcluster fails

diff --git a/taskcluster/kinds/train-student/kind.yml b/taskcluster/kinds/train-student/kind.yml
@@ -57,7 +57,7 @@ tasks:
         worker-type:
             by-tasks-for:
                 github-pull-request: b-largegpu-largedisk
-                default: b-largegpu-xlargedisk
+                default: b-largegpu-xxlargedisk
         worker:
             max-run-time: 2592000
             # train_taskcluster.py exits with 17 if a request to Taskcluster fails