소스 검색

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

myhloli 4 주 전
부모
커밋
f57a6d8d9e
1개의 변경된 파일0개의 추가작업 그리고 1개의 파일을 삭제
  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()