From d2f2aea158698f612288bfff4cf313b660187f8d Mon Sep 17 00:00:00 2001 From: Santiago Castro Date: Thu, 4 Apr 2024 22:49:29 -0400 Subject: [PATCH] Remove an unnecessary list creation in `profiler.py` --- src/training/profiler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/training/profiler.py b/src/training/profiler.py index 1805ca693..4e10a4c26 100644 --- a/src/training/profiler.py +++ b/src/training/profiler.py @@ -123,7 +123,7 @@ def profile_torch(model, text_input_size, image_input_size, batch_size=1, force_ def count_params(model): - return sum([m.numel() for m in model.parameters()]) + return sum(m.numel() for m in model.parameters()) def profile_model(model_name, batch_size=1, profiler='torch'): assert profiler in ['torch', 'fvcore'], 'Only torch and fvcore profilers are supported'