TRASHME test parameters to trigger train in CI

mozilla · May 15, 2024 · 9b3efa9 · 9b3efa9
1 parent 83ff9d7
commit 9b3efa9
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 14 deletions.
diff --git a/taskcluster/translations_taskgraph/parameters.py b/taskcluster/translations_taskgraph/parameters.py
@@ -36,31 +36,31 @@ def get_defaults(_):
             },
             "marian-args": {
                 "training-backward": {
-                    "disp-freq": "1",
-                    "save-freq": "25",
-                    "valid-freq": "50",
+                    "disp-freq": "2",
+                    "save-freq": "26",
+                    "valid-freq": "51",
                     "after": "50u",
                     "dim-vocabs": "1000 1000",
                 },
                 "training-teacher": {
-                    "disp-freq": "1",
-                    "save-freq": "25",
-                    "valid-freq": "50",
+                    "disp-freq": "2",
+                    "save-freq": "26",
+                    "valid-freq": "51",
                     "after": "50u",
                     "dim-vocabs": "1000 1000",
                     "task": "transformer-base",
                 },
                 "training-student": {
-                    "disp-freq": "1",
-                    "save-freq": "25",
-                    "valid-freq": "50",
+                    "disp-freq": "2",
+                    "save-freq": "26",
+                    "valid-freq": "51",
                     "after": "50u",
                     "dim-vocabs": "1000 1000",
                 },
                 "training-student-finetuned": {
-                    "disp-freq": "1",
-                    "save-freq": "25",
-                    "valid-freq": "50",
+                    "disp-freq": "2",
+                    "save-freq": "26",
+                    "valid-freq": "51",
                     "after": "50u",
                     "dim-vocabs": "1000 1000",
                 },
@@ -103,7 +103,7 @@ def get_defaults(_):
                 },
             },
             # Disable Weight & Biases publication on CI
-            "wandb-publication": False,
+            "wandb-publication": True,
         },
     }
 

diff --git a/tracking/translations_parser/wandb.py b/tracking/translations_parser/wandb.py
@@ -89,7 +89,7 @@ def get_wandb_names():
         experiment = {
             "src": "ru",
             "trg": "en",
-            "name": "ci",
+            "name": "ci-evals-pr519",
         }
     else:
         experiment = config["experiment"]