tingwei.gtw yingda.chen 3 years ago
parent
commit
e09d277fd3
1 changed files with 0 additions and 3 deletions
  1. +0
    -3
      modelscope/models/cv/face_human_hand_detection/one_stage_detector.py

+ 0
- 3
modelscope/models/cv/face_human_hand_detection/one_stage_detector.py View File

@@ -56,9 +56,6 @@ class OneStageDetector(nn.Module):


def inference(self, meta): def inference(self, meta):
with torch.no_grad(): with torch.no_grad():
torch.cuda.synchronize()
preds = self(meta['img']) preds = self(meta['img'])
torch.cuda.synchronize()
results = self.head.post_process(preds, meta) results = self.head.post_process(preds, meta)
torch.cuda.synchronize()
return results return results

Loading…
Cancel
Save