diff --git a/nanodet/model/arch/one_stage_detector.py b/nanodet/model/arch/one_stage_detector.py index 90aa3f6f3..8a0f61a81 100644 --- a/nanodet/model/arch/one_stage_detector.py +++ b/nanodet/model/arch/one_stage_detector.py @@ -60,7 +60,7 @@ def inference(self, meta): time2 = time.time() print("forward time: {:.3f}s".format((time2 - time1)), end=" | ") results = self.head.post_process(preds, meta) - + if is_cuda_available: torch.cuda.synchronize()