Update time to ms on CPU

pytorch-labs / segment-anything-fast

A batched offline inference oriented version of segment-anything

Apache License 2.0

1.2k stars 71 forks source link

Update time to ms on CPU #126

Closed yanbing-j closed 3 months ago

yanbing-j commented 4 months ago

This PR is to update the elapsed_time to ms on CPU, aligned with CUDA. start_event.elapsed_time in CUDA is represented by ms, while time.time() is represented by s.

yanbing-j commented 4 months ago

Hi @cpuhrsch , could you please take a look? Thanks!