遍历named_parameters()中的所有的参数,只打印那些param.requires_grad=True的变量。
for name, param in model.named_parameters():
if param.requires_grad:
print(name)
遍历named_parameters()中的所有的参数,只打印那些param.requires_grad=True的变量。
for name, param in model.named_parameters():
if param.requires_grad:
print(name)