pytorch 查看模型参数

对于模型model,可以使用下面的方式打印模型参数:

for name, param in model.named_parameters():
	print(name, '      ', param.size())

你可能感兴趣的:(AI)