Shrink the size of the Teacher in CI

mozilla · Sep 23, 2024 · 2c45b5c · 2c45b5c
1 parent 42639fe
commit 2c45b5c
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/taskcluster/configs/config.ci.yml b/taskcluster/configs/config.ci.yml
@@ -49,7 +49,17 @@ marian-args:
     valid-freq: "50"
     after: 50u
     dim-vocabs: "1000 1000"
+    # Drastically shrink parameters that affect model size:
     task: transformer-base
+    dim-vocabs: "128 128"
+    enc-depth: "2"
+    dec-depth: "2"
+    dim-emb: "16"
+    dim-rnn: "16"
+    transformer-dim-ffn: "128"
+    transformer-dim-aan: "128"
+    transformer-heads: "2"
+    beam-size: "2"
   training-student:
     disp-freq: "1"
     save-freq: "25"