Update tasks/text.py

tasks/text.py

@@ -140,7 +140,7 @@ async def evaluate_text(request: TextEvaluationRequest):
    
     early_stopping = tf.keras.callbacks.EarlyStopping(monitor='val_loss', patience=3, restore_best_weights=True)
  
-    model.fit(train_dataset_bert.shuffle(1000).batch(16),epochs=4,batch_size=16,validation_data=val_dataset_bert.shuffle(1000).batch(16),callbacks=[early_stopping])
+    model.fit(train_dataset_bert.shuffle(1000).batch(16),epochs=3,batch_size=16,validation_data=val_dataset_bert.shuffle(1000).batch(16),callbacks=[early_stopping])
     #--------------------------------------------------------------------------------------------
     # YOUR MODEL INFERENCE STOPS HERE
     #--------------------------------------------------------------------------------------------