fix count

num_params should be directly used.
2026-09-09 11:28:32 +08:00 · 2023-03-08 18:41:30 +08:00
parent 8358b27445
commit d2b99c0b62
1 changed files with 1 additions and 1 deletions
@@ -274,7 +274,7 @@ class PeftModel(PushToHubMixin, torch.nn.Module):

            all_param += num_params
            if param.requires_grad:
-                trainable_params += param.numel()
+                trainable_params += num_params
        print(
            f"trainable params: {trainable_params} || all params: {all_param} || trainable%: {100 * trainable_params / all_param}"
        )