pytorch查看模型参数

作用：查看模型的每一层的参数，我是为了验证预训练的模型参数在每一个epoch中有没有发生改变，而加的下面的代码
代码解析：i代表你想打印的网络层的参数

#------------output model parameters----------------------
i=1
for parameters in model_pretrain.parameters():
    if i==20:
        print(parameters)
        break
    i=i+1