参数相关
参数相关
打印梯度更新的参数
for name, p in model.named_parameters():
if p.requires_grad == True:
print(name)
1
2
3
2
3
统计参数量
sum(p.numel() for p in model.parameters() if p.requires_grad
1
上次更新: 2025/04/02, 12:03:38
Geeks_Z の Blog Geeks_Z の Blog 参数相关for name, p in model.named_parameters():
if p.requires_grad == True:
print(name)
sum(p.numel() for p in model.parameters() if p.requires_grad