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