Skip to content

Commit 2c45b5c

Browse files
committed
Shrink the size of the Teacher in CI
1 parent 42639fe commit 2c45b5c

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

taskcluster/configs/config.ci.yml

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,17 @@ marian-args:
4949
valid-freq: "50"
5050
after: 50u
5151
dim-vocabs: "1000 1000"
52+
# Drastically shrink parameters that affect model size:
5253
task: transformer-base
54+
dim-vocabs: "128 128"
55+
enc-depth: "2"
56+
dec-depth: "2"
57+
dim-emb: "16"
58+
dim-rnn: "16"
59+
transformer-dim-ffn: "128"
60+
transformer-dim-aan: "128"
61+
transformer-heads: "2"
62+
beam-size: "2"
5363
training-student:
5464
disp-freq: "1"
5565
save-freq: "25"

0 commit comments

Comments
 (0)