Skip to content

Commit de861af

Browse files
authored
Update time to ms on CPU (#126)
1 parent 3e9c47d commit de861af

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

experiments/eval_combo.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ def build_results_batch_nested(predictor, batch, batch_size, pad_input_image_bat
103103
torch.cuda.synchronize()
104104
elapsed_time = start_event.elapsed_time(end_event)
105105
else:
106-
elapsed_time = time.time() - t0
106+
elapsed_time = (time.time() - t0) * 1000
107107
return sum(result_batch, []), orig_input_image_batch_size, elapsed_time
108108

109109
def build_results_batch(predictor, batch, batch_size, pad_input_image_batch):
@@ -173,7 +173,7 @@ def build_results_batch(predictor, batch, batch_size, pad_input_image_batch):
173173
torch.cuda.synchronize()
174174
elapsed_time = start_event.elapsed_time(end_event)
175175
else:
176-
elapsed_time = time.time() - t0
176+
elapsed_time = (time.time() - t0) * 1000
177177
return result_batch, orig_input_image_batch_size, elapsed_time
178178

179179

0 commit comments

Comments
 (0)