Pytorch训练过程中改变模型参数 requires_grad 属性

如果模型只在一块GPU上跑,该过程非常简单,只需要训练中途迭代model的 parameters,然后改变各个param的requires_grad 属性即可: for name, param in model.named_parameters(): logger.info('parameter of %s'%name) logger.info('before requires_grad is : %s'%param.requires_grad) param.requires_grad = True logger.info('after requires_gr