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

Signed-off-by: Rui Campos <mail@ruicampos.org>

Author: RuiFilipeCampos

.github/workflows/train-model-sentiment-analysis-task.yml

@@ -54,7 +54,7 @@ jobs:
         MODEL_ATTENTION: ["metric", "scaled_dot_product"]
 
         TRAIN_BATCH_SIZE: [64]
-        TRAIN_EPOCHS: [10]
+        TRAIN_NUMBER_OF_EPOCHS: [10]
         TRAIN_WARMUP_STEPS: [5]
         TRAIN_L1_REGULARIZATION: [0]
         TRAIN_L2_REGULARIZATION: [0]