Explorar el Código

refactor: remove commented-out device assignment in predict_formula.py

myhloli hace 4 semanas
padre
commit
f57a6d8d9e
Se han modificado 1 ficheros con 0 adiciones y 1 borrados
  1. 0 1
      mineru/model/mfr/pp_formulanet_plus_m/predict_formula.py

+ 0 - 1
mineru/model/mfr/pp_formulanet_plus_m/predict_formula.py

@@ -45,7 +45,6 @@ class FormulaRecognizer(BaseOCRV20):
         super(FormulaRecognizer, self).__init__(network_config)
 
         self.load_state_dict(weights)
-        # device = "cpu"
         self.device = torch.device(device) if isinstance(device, str) else device
         self.net.to(self.device)
         self.net.eval()