Change label smoothing to 0.1

Author: ankandrew

fast_plate_ocr/model/custom.py

@@ -61,7 +61,7 @@ def top_3_k(y_true, y_pred):
 
 
 # Custom loss
-def cce_loss(vocabulary_size: int):
+def cce_loss(vocabulary_size: int, label_smoothing: float = 0.1):
     """
     Categorical cross-entropy loss.
     """
@@ -73,7 +73,9 @@ def cce(y_true, y_pred):
         y_true = ops.reshape(y_true, newshape=(-1, vocabulary_size))
         y_pred = ops.reshape(y_pred, newshape=(-1, vocabulary_size))
         return ops.mean(
-            losses.categorical_crossentropy(y_true, y_pred, from_logits=False, label_smoothing=0.2)
+            losses.categorical_crossentropy(
+                y_true, y_pred, from_logits=False, label_smoothing=label_smoothing
+            )
         )
 
     return cce