Merge pull request #162 from dumpmemory/fix_count

fix count
2026-09-09 11:28:32 +08:00 · 2023-03-08 23:03:58 +05:30
parent c1199931de d2b99c0b62
commit 91c69a80ab
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}"
        )