pytorch的model.parameters

  pytorch的model.parameters()的参数指模型中可训练的参数,激活函数、max

pool等是没有训练参数的,如果需固定一些参数不需要,给对应optimizer添加参数的时候需要注意。在model.parameters()返回的结果中,对一个卷积层,权重和偏置各占一个位置。

你可能感兴趣的:(小知识点,pytorch,深度学习,python)