diff --git a/segment_anything_fast/flash_4.py b/segment_anything_fast/flash_4.py index 2124f20..f41905e 100644 --- a/segment_anything_fast/flash_4.py +++ b/segment_anything_fast/flash_4.py @@ -154,7 +154,8 @@ def benchmark_torch_function_in_microseconds(f, *args, **kwargs): t0 = benchmark.Timer( stmt="f(*args, **kwargs)", globals={"args": args, "kwargs": kwargs, "f": f} ) - except: + except Exception as e: + print(f"Error occured: {e}") return None return t0.blocked_autorange().mean * 1e6