[to #42322933] fix cpu inference

修复cpu推理
        Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10468823
This commit is contained in:
tingwei.gtw
2022-10-22 19:19:23 +08:00
committed by yingda.chen
parent 824ee8232c
commit e09d277fd3

View File

@@ -56,9 +56,6 @@ class OneStageDetector(nn.Module):
def inference(self, meta):
with torch.no_grad():
torch.cuda.synchronize()
preds = self(meta['img'])
torch.cuda.synchronize()
results = self.head.post_process(preds, meta)
torch.cuda.synchronize()
return results