作用:查看模型的每一层的参数,我是为了验证预训练的模型参数在每一个epoch中有没有发生改变,而加的下面的代码
代码解析:i代表你想打印的网络层的参数
#------------output model parameters----------------------
i=1
for parameters in model_pretrain.parameters():
if i==20:
print(parameters)
break
i=i+1