Skip to content

Commit

Permalink
fixed one_stage_detector as well for linter
Browse files Browse the repository at this point in the history
  • Loading branch information
cansik committed Dec 30, 2023
1 parent a02e25b commit b87f72c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion nanodet/model/arch/one_stage_detector.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()

Expand Down

0 comments on commit b87f72c

Please sign in to comment.