diff --git a/src/otx/algorithms/segmentation/adapters/mmseg/apis/train.py b/src/otx/algorithms/segmentation/adapters/mmseg/apis/train.py
index f0e4975939d..1e6e34cf2dd 100644
--- a/src/otx/algorithms/segmentation/adapters/mmseg/apis/train.py
+++ b/src/otx/algorithms/segmentation/adapters/mmseg/apis/train.py
@@ -93,9 +93,7 @@ def train_segmentor(model, dataset, cfg, distributed=False, validate=False, time
     if cfg.device == "xpu":
         if cfg.optimizer_config.get("bf16_training", False):
             logger.warning("XPU supports fp32 training only currently.")
-        dtype = torch.float32
         model.train()
-        model, optimizer = torch.xpu.optimize(model, optimizer=optimizer, dtype=dtype)
 
     if "bf16_training" in cfg.optimizer_config:
         # Remove unused parameters in runner