ViT+Lora学习

2026-07-09 21:42:42 +08:00
parent 3efdca98e6
commit 08800b6d37
5 changed files with 474 additions and 1 deletions
@@ -88,7 +88,7 @@ for parm in pretrained_net.layer2.parameters():
 for parm in pretrained_net.layer3.parameters():
    parm.requires_grad=False
 # %%
- def print_trainable_info(model):
+def print_trainable_info(model):
        frozen = sum(p.numel() for p in model.parameters() if not p.requires_grad)
        trainable = sum(p.numel() for p in model.parameters() if p.requires_grad)
        total = frozen + trainable