-
Notifications
You must be signed in to change notification settings - Fork 0
Open
Description
Lines 184 to 188 in 3723b32
| for k, v in model.backbone_layer.named_parameters(): | |
| if "bn" in k: | |
| param_group += [{'params': v, 'lr': args.lr*0.1}] | |
| else: | |
| v.requires_grad = False |
Hello, thanks for sharing your code. In the above a few lines, only parameters in BN layers will be updated and all the other parameters are frozen. Does this mean only BN parameters are trained?
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels