style: apply linter

illuin-tech · Sep 27, 2024 · ba15d9c · ba15d9c
1 parent 2956a4d
commit ba15d9c
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/colpali_engine/utils/torch_utils.py b/colpali_engine/utils/torch_utils.py
@@ -62,6 +62,5 @@ def print_trainable_parameters(model: nn.Module) -> None:
         all_param += param.numel()
         if param.requires_grad:
             trainable_params += param.numel()
-    print(
-        f"trainable params: {trainable_params:,} || all params: {all_param:,} || trainable%: {100 * trainable_params / all_param}"
-    )
+    trainable_percentage = 100 * trainable_params / all_param
+    print(f"trainable params: {trainable_params:,} || all params: {all_param:,} || trainable%: {trainable_percentage}")