fix count

num_params should be directly used.
This commit is contained in:
dumpmemory authored and GitHub committed 2023-03-08 18:41:30 +08:00
1 parent 8358b27445
commit d2b99c0b62
1 file changed
+1 -1
+1 -1
View File
@@ -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}"
)