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