Эх сурвалжийг харах

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

myhloli 4 долоо хоног өмнө
parent
commit
f57a6d8d9e

+ 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()