Remove torch.xpu.optimize for semantic_segmentation task (#3172)

* remove torch.optimize * fix pre-commit
openvinotoolkit · Apr 9, 2024 · a999372 · a999372
1 parent 4ce3e40
commit a999372
Showing 1 changed file with 0 additions and 2 deletions.
diff --git 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