Update train-model-sentiment-analysis-task.yml

Signed-off-by: Rui Campos <mail@ruicampos.org>
Digital-Defiance · Jan 30, 2024 · 9046cc2 · 9046cc2
1 parent c5ffac9
commit 9046cc2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/train-model-sentiment-analysis-task.yml b/.github/workflows/train-model-sentiment-analysis-task.yml
@@ -70,7 +70,7 @@ jobs:
         MODEL_ATTENTION: ${{ matrix.MODEL_ATTENTION }}
 
         TRAIN_BATCH_SIZE: ${{ matrix.TRAIN_BATCH_SIZE }}
-        TRAIN_EPOCHS: ${{ matrix.TRAIN_EPOCHS }}
+        TRAIN_NUMBER_OF_EPOCHS: ${{ matrix.TRAIN_NUMBER_OF_EPOCHS }}
         TRAIN_L1_REGULARIZATION: ${{ matrix.TRAIN_L1_REGULARIZATION }}
         TRAIN_L2_REGULARIZATION: ${{ matrix.TRAIN_L2_REGULARIZATION }}
         TRAIN_WARMUP_STEPS: ${{ matrix.TRAIN_WARMUP_STEPS }}